Fix LightGBM missing argument in predict method

This commit is contained in:
Matthias 2022-07-10 11:05:35 +02:00
parent 2e1061af64
commit 58b18770e3

View File

@ -91,7 +91,7 @@ class LightGBMPredictionModel(IFreqaiModel):
return model return model
def predict( def predict(
self, unfiltered_dataframe: DataFrame, dk: FreqaiDataKitchen self, unfiltered_dataframe: DataFrame, dk: FreqaiDataKitchen, first: bool = False
) -> Tuple[DataFrame, DataFrame]: ) -> Tuple[DataFrame, DataFrame]:
""" """
Filter the prediction features data and predict with it. Filter the prediction features data and predict with it.