ninth commit

perpetual-ml · Jun 7, 2024 · 97cda78 · 97cda78
1 parent d0ef036
commit 97cda78
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -55,7 +55,7 @@ jobs:
   macos-build-test:
     strategy:
       matrix:
-        pyversion: ["3.8", "3.9", "3.10", "3.11", "3.12"]
+        pyversion: ["3.11", "3.12"]
     runs-on: "macos-latest"
     steps:
       - uses: actions/checkout@v4

diff --git a/python-package/python/perpetual/__init__.py b/python-package/python/perpetual/__init__.py
@@ -178,7 +178,7 @@ def convert_input_frame(X: FrameLike, categorical_features) -> tuple[list[str],
             def f(x):
                 try:
                     return np.nan if str(x[feature_index]) == "nan" else float(categories.index(str(x[feature_index])))
-                except:
+                except (ValueError, IndexError):
                     return np.nan
             X_[:, feature_index] = np.apply_along_axis(f, 1, X_)
 
@@ -215,7 +215,7 @@ def transform_input_frame(X: FrameLike, cat_mapping) -> tuple[list[str], np.ndar
             def f(x):
                 try:
                     return np.nan if str(x[feature_index]) == "nan" else float(categories.index(str(x[feature_index])))
-                except:
+                except (ValueError, IndexError):
                     return np.nan
             X_[:, feature_index] = np.apply_along_axis(f, 1, X_)
 

diff --git a/python-package/tests/test_booster.py b/python-package/tests/test_booster.py
@@ -34,7 +34,6 @@ def test_booster_no_variance(X_y):
 
     model = PerpetualBooster(objective="LogLoss")
     model.fit(X, y)
-    preds = model.predict(X)
     assert model.feature_importances_[1] == 0.0
     assert model.feature_importances_[3] == 0.0