From 8a2afeb23c233bea9bae9c41d3ab74124be992bc Mon Sep 17 00:00:00 2001 From: himkwtn Date: Thu, 5 Sep 2024 11:52:52 -0700 Subject: [PATCH] fix lint --- pysindy/utils/base.py | 6 +++--- test/utils/test_utils.py | 1 - 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/pysindy/utils/base.py b/pysindy/utils/base.py index a1be22b6..3d5c9ee3 100644 --- a/pysindy/utils/base.py +++ b/pysindy/utils/base.py @@ -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 """ @@ -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 """ diff --git a/test/utils/test_utils.py b/test/utils/test_utils.py index 5c3e32a7..252ebc31 100644 --- a/test/utils/test_utils.py +++ b/test/utils/test_utils.py @@ -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):