fix possible memory leak associated with Catboost Pool object

2022-07-26 17:29:29 +02:00
parent 3f149c4067
commit 324e54c015
4 changed files with 11 additions and 8 deletions
--- a/freqtrade/freqai/prediction_models/BaseRegressionModel.py
+++ b/freqtrade/freqai/prediction_models/BaseRegressionModel.py
@@ -18,7 +18,7 @@ class BaseRegressionModel(IFreqaiModel):
    such as prediction_models/CatboostPredictionModel.py for guidance.
    """

-    def return_values(self, dataframe: DataFrame, dk: FreqaiDataKitchen) -> DataFrame:
+    def return_values(self, dataframe: DataFrame) -> DataFrame:
        """
        User uses this function to add any additional return values to the dataframe.
        e.g.