Skip to content

Commit 1aba9e6

Browse files
committed
fix bug
1 parent 77954f3 commit 1aba9e6

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

stemflow/model/SphereAdaSTEM.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -333,7 +333,8 @@ def split(
333333
else:
334334
self.ensemble_df, self.gridding_plot = ensemble_df, None
335335

336-
def SAC_ensemble_training(self, single_ensemble_df: pd.DataFrame, X_train: Union[pd.DataFrame, str], y_train: Union[pd.DataFrame, str]):
336+
def SAC_ensemble_training(self, single_ensemble_df: pd.DataFrame, X_train: Union[pd.DataFrame, str], y_train: Union[pd.DataFrame, str],
337+
temporal_window_prequery: bool = False):
337338
"""A sub-module of SAC training function.
338339
Train only one ensemble.
339340

0 commit comments

Comments
 (0)