From 2c4e5d79ac7a6887ddcd067de9d46faae2d9936d Mon Sep 17 00:00:00 2001 From: =?utf8?q?J=C3=A9r=C3=B4me=20Benoit?= Date: Sun, 4 Jan 2026 00:01:16 +0100 Subject: [PATCH] refactor(quickadapter): code cleanups and optimizations MIME-Version: 1.0 Content-Type: text/plain; charset=utf8 Content-Transfer-Encoding: 8bit Signed-off-by: Jérôme Benoit --- quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py | 1 + .../user_data/strategies/ExtremaWeightingTransformer.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py index 11688f2..917a009 100644 --- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py +++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py @@ -2973,6 +2973,7 @@ class QuickAdapterRegressorV3(BaseRegressionModel): f"Supported: {', '.join(QuickAdapterRegressorV3._OPTUNA_SAMPLERS)}" ) + @lru_cache(maxsize=8) def optuna_samplers_by_namespace( self, namespace: OptunaNamespace ) -> tuple[tuple[OptunaSampler, ...], OptunaSampler]: diff --git a/quickadapter/user_data/strategies/ExtremaWeightingTransformer.py b/quickadapter/user_data/strategies/ExtremaWeightingTransformer.py index 7a611be..a2b76ca 100644 --- a/quickadapter/user_data/strategies/ExtremaWeightingTransformer.py +++ b/quickadapter/user_data/strategies/ExtremaWeightingTransformer.py @@ -208,7 +208,7 @@ class ExtremaWeightingTransformer(BaseTransform): if np.isclose(gamma, 1.0) or not np.isfinite(gamma) or gamma <= 0: return values out = values.copy() - out[mask] = np.power(np.abs(values[mask]), 1.0 / gamma) * np.sign(values[mask]) + out[mask] = np.sign(values[mask]) * np.power(np.abs(values[mask]), 1.0 / gamma) return out def fit( -- 2.43.0