diff --git a/src/nemos/basis/_basis_mixin.py b/src/nemos/basis/_basis_mixin.py index 21ade27d..79c882c2 100644 --- 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(),