Fix LightGBM missing argument in predict method
This commit is contained in:
parent
2e1061af64
commit
58b18770e3
@ -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.
|
||||
|
Loading…
Reference in New Issue
Block a user