Fix LightGBM missing argument in predict method

2022-07-10 11:05:35 +02:00
parent 2e1061af64
commit 58b18770e3
1 changed files with 1 additions and 1 deletions
--- a/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py
+++ b/freqtrade/freqai/prediction_models/LightGBMPredictionModel.py
@@ -91,7 +91,7 @@ class LightGBMPredictionModel(IFreqaiModel):
        return model

    def predict(
-        self, unfiltered_dataframe: DataFrame, dk: FreqaiDataKitchen
+        self, unfiltered_dataframe: DataFrame, dk: FreqaiDataKitchen, first: bool = False
    ) -> Tuple[DataFrame, DataFrame]:
        """
        Filter the prediction features data and predict with it.