Skip to content

Commit 5e1ad04

Browse files
committed
fix fialing test due to fista not initializing datafit
1 parent fb4cd15 commit 5e1ad04

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

skglm/solvers/fista.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,10 +51,12 @@ def _solve(self, X, y, datafit, penalty, w_init=None, Xw_init=None):
5151
Xw = Xw_init.copy() if Xw_init is not None else np.zeros(n_samples)
5252

5353
if X_is_sparse:
54+
datafit.initialize_sparse(X.data, X.indptr, X.indices, y)
5455
lipschitz = datafit.get_global_lipschitz_sparse(
5556
X.data, X.indptr, X.indices, y
5657
)
5758
else:
59+
datafit.initialize(X, y)
5860
lipschitz = datafit.get_global_lipschitz(X, y)
5961

6062
for n_iter in range(self.max_iter):

0 commit comments

Comments
 (0)