🐛 Small fix

Author: gozdeg

tiatoolbox/models/engine/engine_abc.py

@@ -381,7 +381,7 @@ def _get_model_attr(self: EngineABC, attr_name: str) -> Callable:
             return getattr(self.model, attr_name)
         except AttributeError:
             module = getattr(self.model, "module", None)
-            return getattr(module, attr_name, None)
+            return getattr(module, attr_name)
 
     def get_dataloader(
         self: EngineABC,