didn't mean to remove that...

flatironinstitute · Nov 20, 2023 · 5cfdffc · 5cfdffc
1 parent 6ae6c8e
commit 5cfdffc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nemos/basis.py b/src/nemos/basis.py
@@ -64,7 +64,7 @@ def evaluate(self, *xi: NDArray) -> NDArray:
         pass
 
     @staticmethod
-    def _get_samples(*n_samples: int):
+    def _get_samples(*n_samples: int) -> Generator[NDArray, ...]:
         """Get equi-spaced samples for all the input dimensions.
 
         This will be used to evaluate the basis on a grid of