improved decorator logic

src/nemos/basis/_basis.py

@@ -37,7 +37,11 @@ def wrapper(self, **params):
         current_params = self.__sklearn_get_params__()
         for key, val in new_params.items():
             current = current_params[key]
-            if isinstance(val, Basis) and val._has_default_label and not current._has_default_label:
+            if (
+                isinstance(val, Basis)
+                and val._has_default_label
+                and not current._has_default_label
+            ):
                 try:
                     val.label = current.label
                 except ValueError:
@@ -367,7 +371,6 @@ def get_params(self, deep=True) -> dict:
     def set_params(self, **params: Any):
         return super().set_params(**params)
 
-
     @property
     def n_output_features(self) -> int | None:
         """

src/nemos/basis/_basis_mixin.py

@@ -6,7 +6,6 @@
 import copy
 import inspect
 import re
-import sys
 from functools import wraps
 from itertools import chain
 from typing import TYPE_CHECKING, Generator, List, Literal, Optional, Tuple, Union
@@ -39,7 +38,7 @@
     "OrthExponentialEval",
     "OrthExponentialConv",
     "AdditiveBasis",
-    "MultiplicativeBasis"
+    "MultiplicativeBasis",
 ]
 
 
@@ -165,7 +164,6 @@ def label(self, label: str | None) -> None:
     def _has_default_label(self):
         return re.match(rf"^{self.__class__.__name__}(_\d+)?$", self._label) is not None
 
-
     def _recompute_all_labels(self):
         """
         Recompute all labels matching default for self.

src/nemos/basis/basis.py

@@ -9,7 +9,12 @@
 
 from ..typing import FeatureMatrix
 from ._basis import add_docstring
-from ._basis_mixin import AtomicBasisMixin, ConvBasisMixin, EvalBasisMixin, __PUBLIC_BASES__
+from ._basis_mixin import (
+    __PUBLIC_BASES__,
+    AtomicBasisMixin,
+    ConvBasisMixin,
+    EvalBasisMixin,
+)
 from ._decaying_exponential import OrthExponentialBasis
 from ._identity import HistoryBasis, IdentityBasis
 from ._raised_cosine_basis import RaisedCosineBasisLinear, RaisedCosineBasisLog