silent bug multitask

scikit-learn-contrib · Jun 2, 2024 · 165ad25 · 165ad25
1 parent 9a3b940
commit 165ad25
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/skglm/solvers/multitask_bcd.py b/skglm/solvers/multitask_bcd.py
@@ -66,7 +66,9 @@ def solve(self, X, Y, datafit, penalty, W_init=None, XW_init=None):
             if self.ws_strategy == "subdiff":
                 opt = penalty.subdiff_distance(W, grad, all_feats)
             elif self.ws_strategy == "fixpoint":
-                opt = dist_fix_point_bcd(W, grad, datafit, penalty, all_feats)
+                opt = dist_fix_point_bcd(
+                    W, grad, lipschitz, datafit, penalty, all_feats
+                )
             stop_crit = np.max(opt)
             if self.verbose:
                 print(f"Stopping criterion max violation: {stop_crit:.2e}")