def _can_register_hook()

in captum/attr/_core/deep_lift.py [0:0]

• 8 lines of code
• 5 McCabe index (conditional complexity)

    def _can_register_hook(self, module: Module) -> bool:
        # TODO find a better way of checking if a module is a container or not
        module_fullname = str(type(module))
        has_already_hooks = len(module._backward_hooks) > 0  # type: ignore
        return not (
            "nn.modules.container" in module_fullname
            or has_already_hooks
            or not self._is_non_linear(module)
        )