Skip to content

Commit 49c3444

Browse files
author
SamoraHunter
committed
disabled multithread for runner
1 parent 4ae75f9 commit 49c3444

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

ml_grid/model_classes/H2OBaseClassifier.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -521,7 +521,7 @@ def predict(self, X: pd.DataFrame) -> np.ndarray:
521521
raise RuntimeError(f"H2O prediction failed: {e}")
522522

523523
# Extract predictions
524-
pred_df = predictions.as_data_frame(use_multi_thread=True)
524+
pred_df = predictions.as_data_frame(use_multi_thread=False)
525525
if "predict" in pred_df.columns:
526526
return pred_df["predict"].values.ravel()
527527
else:
@@ -580,7 +580,7 @@ def predict_proba(self, X: pd.DataFrame) -> np.ndarray:
580580
raise RuntimeError(f"H2O prediction failed: {e}")
581581

582582
# Extract probabilities (drop the 'predict' column)
583-
prob_df = predictions.drop("predict").as_data_frame(use_multi_thread=True)
583+
prob_df = predictions.drop("predict").as_data_frame(use_multi_thread=False)
584584
return prob_df.values
585585

586586
def _ensure_model_is_loaded(self):

0 commit comments

Comments
 (0)