forgot some

Quantco · Feb 19, 2025 · 533d536 · 533d536
1 parent 923bbdc
commit 533d536
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/glum_benchmarks/bench_liblinear.py b/src/glum_benchmarks/bench_liblinear.py
@@ -92,7 +92,7 @@ def liblinear_bench(
         solver="liblinear",
     )
 
-    fit_args = dict(
+    fit_args = dict(  # type: ignore
         X=X,
         y=dat["y"].astype(np.int64).copy(),
         sample_weight=dat.get("sample_weight"),

diff --git a/src/glum_benchmarks/util.py b/src/glum_benchmarks/util.py
@@ -108,8 +108,8 @@ def get_obj_val(
         X_dot_coef += offset
 
     zeros = np.zeros(dat["X"].shape[0])
-    y = dat["y"].astype(coefs.dtype)
-    weights = dat.get("weights", np.ones_like(y)).astype(coefs.dtype)
+    y = dat["y"].astype(coefs.dtype)  # type: ignore
+    weights = dat.get("weights", np.ones_like(y)).astype(coefs.dtype)  # type: ignore
     weights /= weights.sum()
     P1 = l1_ratio * alpha * np.ones_like(coefs)
     P2 = (1 - l1_ratio) * alpha * np.ones_like(coefs)