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

1 year ago · a8ae767576
parent b9c77124fc
commit a8ae767576
1 changed files with 2 additions and 1 deletions
--- a/freqtrade/optimize/hyperopt/hyperopt_optimizer.py
+++ b/freqtrade/optimize/hyperopt/hyperopt_optimizer.py
@ -122,7 +122,8 @@ class HyperOptimizer:
        """
        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(