change docstrings

flatironinstitute · Dec 15, 2024 · 28599c8 · 28599c8
1 parent 3d91df1
commit 28599c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/nemos/basis/_basis_mixin.py b/src/nemos/basis/_basis_mixin.py
@@ -96,7 +96,7 @@ def _iterate_over_components(self) -> Generator:
 
         Returns
         -------
-            A list with the basis components.
+            A generator returning self, it will be chained for composite bases.
 
         """
         return (x for x in [self])
@@ -608,7 +608,7 @@ def _iterate_over_components(self):
 
         Returns
         -------
-            A list with all 1d basis components.
+            A generator looping on each individual input.
         """
         return chain(
             self._basis1._iterate_over_components(),