🐞 Bug Fix: Return results of load_state_dict func (#546)

you have to reutrn the results of load_state_dict if you wrappered the original one
openvinotoolkit · Sep 9, 2022 · dff86b1 · dff86b1
1 parent c37764c
commit dff86b1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/anomalib/models/components/base/anomaly_module.py b/anomalib/models/components/base/anomaly_module.py
@@ -189,4 +189,4 @@ def load_state_dict(self, state_dict: OrderedDict[str, Tensor], strict: bool = T
         """
         # Used to load missing normalization and threshold parameters
         self._load_normalization_class(state_dict)
-        super().load_state_dict(state_dict, strict=strict)
+        return super().load_state_dict(state_dict, strict=strict)