Skip to content

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
himkwtn committed Sep 5, 2024
1 parent a0475aa commit 8a2afeb
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 4 deletions.
6 changes: 3 additions & 3 deletions pysindy/utils/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -186,8 +186,8 @@ def get_prox(
Returns:
--------
proximal_operator: (x: np.array, reg_weight: float | np.array) -> np.array
A function that takes an input array x and a regularization weight,
which can be either a scalar or array of the same shape,
A function that takes an input array x and a regularization weight,
which can be either a scalar or array of the same shape,
and returns an array of the same shape
"""

Expand Down Expand Up @@ -234,7 +234,7 @@ def get_regularization(
Returns:
--------
regularization_function: (x: np.array, reg_weight: float | np.array) -> np.array
A function that takes an input array x and a regularization weight,
A function that takes an input array x and a regularization weight,
which can be either a scalar or array of the same shape,
and returns a float
"""
Expand Down
1 change: 0 additions & 1 deletion test/utils/test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,6 @@ def test_get_regularization(regularization, lam, expected):
[
np.array([[1, 2]]),
np.array([[1]]),
],
)
def test_get_prox_and_regularization_bad_shape(regularization, lam):
Expand Down

0 comments on commit 8a2afeb

Please sign in to comment.