From 3d9b18764e4b619598f3edfeeddc378a47a3b711 Mon Sep 17 00:00:00 2001 From: himkwtn Date: Wed, 11 Sep 2024 12:25:41 -0700 Subject: [PATCH] fix class params in tests --- pysindy/optimizers/stable_linear_sr3.py | 2 +- test/test_feature_library.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/pysindy/optimizers/stable_linear_sr3.py b/pysindy/optimizers/stable_linear_sr3.py index 9742e2e1..069f1a1b 100644 --- a/pysindy/optimizers/stable_linear_sr3.py +++ b/pysindy/optimizers/stable_linear_sr3.py @@ -162,7 +162,7 @@ def _create_var_and_part_cost( 2 * self.relax_coeff_nu ) penalty = self._calculate_penalty( - self.thresholder, np.ravel(self.reg_weight_lam), xi + self.regularizer, np.ravel(self.reg_weight_lam), xi ) return xi, cost + penalty diff --git a/test/test_feature_library.py b/test/test_feature_library.py index 4740db6a..8ba7516a 100644 --- a/test/test_feature_library.py +++ b/test/test_feature_library.py @@ -701,7 +701,7 @@ def test_sindypi_library(data_lorenz): function_names=library_function_names, include_bias=True, ) - sindy_opt = SINDyPI(threshold=0.1, thresholder="l1") + sindy_opt = SINDyPI(reg_weight_lam=0.1, regularizer="l1") model = SINDy( optimizer=sindy_opt, feature_library=sindy_library, @@ -709,7 +709,7 @@ def test_sindypi_library(data_lorenz): model.fit(x, t=t) assert np.shape(sindy_opt.coef_) == (40, 40) - sindy_opt = SINDyPI(threshold=1, thresholder="l1", model_subset=[3]) + sindy_opt = SINDyPI(reg_weight_lam=1, regularizer="l1", model_subset=[3]) model = SINDy( optimizer=sindy_opt, feature_library=sindy_library,