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/BaseTensorFlowModel.py
+++ b/freqtrade/freqai/prediction_models/BaseTensorFlowModel.py
@@ -16,7 +16,7 @@ class BaseTensorFlowModel(IFreqaiModel):
    User *must* inherit from this class and set fit() and predict().
    """

-    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.