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