diff --git a/src/nemos/basis/_basis_mixin.py b/src/nemos/basis/_basis_mixin.py index 4d6e8440..ba7205d0 100644 --- a/src/nemos/basis/_basis_mixin.py +++ b/src/nemos/basis/_basis_mixin.py @@ -85,7 +85,7 @@ def __sklearn_clone__(self) -> Basis: """Clone the basis while preserving attributes related to input shapes. This method ensures that input shape attributes (e.g., `_input_shape_product`, - `_input_shape_product`) are preserved during cloning. Reinitializing the class + `_input_shape_`) are preserved during cloning. Reinitializing the class as in the regular sklearn clone would drop these attributes, rendering cross-validation unusable. """