CLN: fix linting

dynamicslab · Aug 26, 2024 · 6928455 · 6928455
1 parent ba3798e
commit 6928455
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/pysindy/optimizers/constrained_sr3.py b/pysindy/optimizers/constrained_sr3.py
@@ -276,8 +276,10 @@ def _calculate_penalty(
         """
         Args:
         -----
-        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)
+        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
 
         Returns:

diff --git a/pysindy/utils/base.py b/pysindy/utils/base.py
@@ -191,7 +191,7 @@ def get_prox(
     --------
     proximal_operator: (x: np.array, reg_weight: float | np.array) -> np.array
         A function that takes an input x of shape (n_targets, n_features)
-        and regularization weight factor which can be a scalar or 
+        and regularization weight factor which can be a scalar or
         an array of shape (n_targets, n_features),
         and returns an array of shape (n_targets, n_features)
     """
@@ -252,7 +252,7 @@ def get_regularization(
     --------
     regularization_function: (x: np.array, reg_weight: float | np.array) -> np.array
         A function that takes an input x of shape (n_targets, n_features)
-        and regularization weight factor which can be a scalar or 
+        and regularization weight factor which can be a scalar or
         an array of shape (n_targets, n_features),
         and returns a float
     """