CLN: fix constrained SR3 docstring

dynamicslab · Aug 30, 2024 · 5ee3bcc · 5ee3bcc
1 parent c2237bf
commit 5ee3bcc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysindy/optimizers/constrained_sr3.py b/pysindy/optimizers/constrained_sr3.py
@@ -279,8 +279,8 @@ def _calculate_penalty(
         regularization: 'l0' | 'weighted_l0' | 'l1' | 'weighted_l1' |
                         'l2' | 'weighted_l2'
         regularization_weight: float | np.array, can be a scalar
-                               or an array of shape (n_targets, n_features)
-        xi: cp.Variable
+                               or an array of shape (m, n)
+        xi: cp.Variable of length m*n
 
         Returns:
         --------