update

chenyangkang · Nov 20, 2024 · 699c14c · 699c14c
1 parent 226abd2
commit 699c14c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 16 deletions.
diff --git 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
@@ -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")