Merge pull request #8216 from freqtrade/close-rl-env

Close training and eval environments in ReinforcementLearner_multiproc
2023-02-26 15:20:32 +01:00 · 2023-02-26 15:20:32 +01:00 · d3d7cb1b14
commit d3d7cb1b14
parent 305eda74e2 cb80d7c26f
1 changed files with 5 additions and 0 deletions
--- a/freqtrade/freqai/prediction_models/ReinforcementLearner_multiproc.py
+++ b/freqtrade/freqai/prediction_models/ReinforcementLearner_multiproc.py
@ -34,6 +34,11 @@ class ReinforcementLearner_multiproc(ReinforcementLearner):
        train_df = data_dictionary["train_features"]
        test_df = data_dictionary["test_features"]

+        if self.train_env:
+            self.train_env.close()
+        if self.eval_env:
+            self.eval_env.close()
+
        env_info = self.pack_env_dict(dk.pair)

        env_id = "train_env"