def __init__()

in code/pretrained_model_with_debugger_hook.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, model: Any, hook: Union[smd.Hook, None]):
        '''Simple container to associate a 'model' with a SageMaker debug hook'''
        self.model = model
        self.hook = hook