From 699c14cacfb8577803134d646d83460a57b5a1a9 Mon Sep 17 00:00:00 2001 From: chenyangkang Date: Tue, 19 Nov 2024 18:12:46 -0600 Subject: [PATCH] update --- .github/workflows/run_pytest.yml | 2 +- stemflow/model/static_func_AdaSTEM.py | 15 --------------- 2 files changed, 1 insertion(+), 16 deletions(-) diff --git a/.github/workflows/run_pytest.yml b/.github/workflows/run_pytest.yml index 735bf93..96002e2 100644 --- a/.github/workflows/run_pytest.yml +++ b/.github/workflows/run_pytest.yml @@ -33,7 +33,7 @@ jobs: - name: Run pytest run: | - pytest -n auto --cov --no-cov-on-fail --cov-report=term-missing:skip-covered --cov-report xml:coverage.xml + pytest -n auto --cov --no-cov-on-fail --cov-report=term-missing:skip-covered --cov-report xml:coverage.xml -v timeout-minutes: 300 # Set the timeout to 1.5 hours for this step - name: Upload coverage reports to Codecov diff --git a/stemflow/model/static_func_AdaSTEM.py b/stemflow/model/static_func_AdaSTEM.py index 3b6e73a..112704a 100644 --- a/stemflow/model/static_func_AdaSTEM.py +++ b/stemflow/model/static_func_AdaSTEM.py @@ -96,23 +96,8 @@ def train_one_stixel( trained_model.fit(sub_X_train[stixel_specific_x_names], sub_y_train, sample_weight=sample_weights) trained_model.my_class_weights = class_weights - # try: - # trained_model.fit(sub_X_train[stixel_specific_x_names], sub_y_train, sample_weight=sample_weights) - - # except Exception as e: - # print(e) - # # raise - # return (None, [], "Base_model_fitting_error(non-regression, balanced weight)") else: trained_model.fit(sub_X_train[stixel_specific_x_names], sub_y_train) - - # try: - # trained_model.fit(sub_X_train[stixel_specific_x_names], sub_y_train) - - # except Exception as e: - # print(e) - # # raise - # return (None, [], "Base_model_fitting_error(regression)") return (trained_model, stixel_specific_x_names, "Success")