linted

flatironinstitute · Nov 29, 2024 · cbfc4a7 · cbfc4a7
1 parent db72fda
commit cbfc4a7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/nemos/basis/basis.py b/src/nemos/basis/basis.py
@@ -1059,7 +1059,12 @@ def split_by_feature(
         >>> import numpy as np
         >>> from nemos.basis import ConvOrthExponential
         >>> from nemos.glm import GLM
-        >>> basis = ConvOrthExponential(n_basis_funcs=6, decay_rates=np.arange(1, 7), window_size=10, label="two_inputs")
+        >>> basis = ConvOrthExponential(
+        ...     n_basis_funcs=6,
+        ...     decay_rates=np.arange(1, 7),
+        ...     window_size=10,
+        ...     label="two_inputs"
+        ... )
         >>> X_multi = basis.compute_features(np.random.randn(20, 2))
         >>> split_features_multi = basis.split_by_feature(X_multi, axis=1)
         >>> for feature, sub_dict in split_features_multi.items():