fix

chenyangkang · Nov 20, 2024 · a63a0d2 · a63a0d2
1 parent 88b795c
commit a63a0d2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/run_pytest.yml b/.github/workflows/run_pytest.yml
@@ -9,7 +9,7 @@ jobs:
   pytesting:
     runs-on: ubuntu-20.04
     strategy:
-      fail-fast: false
+      fail-fast: true
       matrix:
         python-version: ["3.8", "3.9", "3.10"]
     steps:

diff --git a/tests/test_model.py b/tests/test_model.py
@@ -281,7 +281,7 @@ def test_SphereAdaSTEMRegressor():
 
 def test_AdaSTEMRegressor_Hurdle_for_AdaSTEM():
     model = make_AdaSTEMRegressor_Hurdle_for_AdaSTEM()
-    model = model.fit(X_train, np.where(y_train > 0, 1, 0))
+    model = model.fit(X_train, y_train)
 
     pred_mean = model.predict(X_test.reset_index(drop=True))
     assert np.sum(~np.isnan(pred_mean)) > 0