Pre-commit hooks (#732)

Quantco · Nov 13, 2023 · b14c524 · b14c524
1 parent a304d55
commit b14c524
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/glum/_solvers.py b/src/glum/_solvers.py
@@ -758,7 +758,7 @@ def line_search(state: IRLSState, data: IRLSData, d: np.ndarray):
     """
     # line search parameters
     (beta, sigma) = (0.5, 0.0001)
-    eps = 16 * np.finfo(state.obj_val.dtype).eps
+    eps = 16 * np.finfo(state.obj_val.dtype).eps  # type: ignore
 
     # line search by sequence beta^k, k=0, 1, ..
     # F(w + lambda d) - F(w) <= lambda * bound

diff --git a/src/glum_benchmarks/data/create_insurance.py b/src/glum_benchmarks/data/create_insurance.py
@@ -6,7 +6,6 @@
 import pandas as pd
 import sklearn.compose
 from git_root import git_root
-from pandas.api.types import is_categorical_dtype
 from sklearn.base import BaseEstimator, TransformerMixin
 from sklearn.compose._column_transformer import _get_transformer_list
 from sklearn.pipeline import Pipeline
@@ -109,7 +108,7 @@ def fit(
         categories = {}
         for name in columns:
             col = X[name]
-            if not is_categorical_dtype(col):
+            if str(col.dtype) != "category":  # type: ignore
                 col = pd.Series(col, index=X.index).astype("category")
             categories[name] = col.dtype