Skip to content

Commit 0facfd7

Browse files
committed
fix some tests
1 parent e8004ba commit 0facfd7

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

skglm/solvers/base.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,10 @@ def solve(self, X, y, datafit, penalty, w_init=None, Xw_init=None,
110110
warnings.warn(
111111
"Do not pass a compiled datafit, compilation is done inside solver now")
112112
else:
113-
datafit = compiled_clone(datafit, to_float32=X.dtype == np.float32)
114-
penalty = compiled_clone(penalty)
113+
if datafit is not None:
114+
datafit = compiled_clone(datafit, to_float32=X.dtype == np.float32)
115+
if penalty is not None:
116+
penalty = compiled_clone(penalty)
115117
# TODO add support for bool spec in compiled_clone
116118
# penalty = compiled_clone(penalty, to_float32=X.dtype == np.float32)
117119

0 commit comments

Comments
 (0)