From a63a0d2fe452948594fe2e77acba098eeb1e8f2d Mon Sep 17 00:00:00 2001 From: chenyangkang Date: Tue, 19 Nov 2024 23:10:59 -0600 Subject: [PATCH] fix --- .github/workflows/run_pytest.yml | 2 +- tests/test_model.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run_pytest.yml b/.github/workflows/run_pytest.yml index 7431c29..369d66e 100644 --- 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 index 3968e49..6a64532 100644 --- 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