fix: Hyperopt-magic - allow parent classes within the same file

Uncodedtech · Feb 9, 2025 · a8ae767 · a8ae767
1 parent b9c7712
commit a8ae767
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/freqtrade/optimize/hyperopt/hyperopt_optimizer.py b/freqtrade/optimize/hyperopt/hyperopt_optimizer.py
@@ -122,7 +122,8 @@ def hyperopt_pickle_magic(self, bases: tuple[type, ...]) -> None:
         """
         for modules in bases:
             if modules.__name__ != "IStrategy":
-                cloudpickle.register_pickle_by_value(sys.modules[modules.__module__])
+                if mod := sys.modules.get(modules.__module__):
+                    cloudpickle.register_pickle_by_value(mod)
                 self.hyperopt_pickle_magic(modules.__bases__)
 
     def _get_params_dict(