Skip to content

Commit c08b6ac

Browse files
committed
comments and TODOs
1 parent b838317 commit c08b6ac

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

skglm/solvers/base.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,17 +107,23 @@ def solve(
107107
>>> ...
108108
>>> coefs, obj_out, stop_crit = solver.solve(X, y, datafit, penalty)
109109
"""
110+
# TODO do it properly instead of searching for a string
110111
if "jitclass" in str(type(datafit)):
111112
warnings.warn(
112113
"Do not pass a compiled datafit, compilation is done inside solver now"
113114
)
115+
if "jitclass" in str(type(penalty)):
116+
warnings.warn(
117+
"Do not pass a compiled penalty, compilation is done inside solver now"
118+
)
114119
else:
115120
if datafit is not None:
116121
datafit = compiled_clone(datafit, to_float32=X.dtype == np.float32)
117122
if penalty is not None:
118123
penalty = compiled_clone(penalty)
119-
# TODO add support for bool spec in compiled_clone
120-
# penalty = compiled_clone(penalty, to_float32=X.dtype == np.float32)
124+
# TODO add support for bool spec in compiled_clone
125+
# currently, doing so break the code
126+
# penalty = compiled_clone(penalty, to_float32=X.dtype == np.float32)
121127

122128
if run_checks:
123129
self._validate(X, y, datafit, penalty)

0 commit comments

Comments
 (0)