Initial commit

1 files changed, 13 insertions, 0 deletions

diff --git a/sbdata/util.py b/sbdata/util.py
new file mode 100644
index 0000000..87a3098
--- /dev/null
+++ b/sbdata/util.py
@@ -0,0 +1,13 @@
+import functools
+from typing import TypeVar, Generator, ParamSpec, Callable
+
+_Param = ParamSpec('_Param')
+_RetType = TypeVar('_RetType')
+
+
+def no_generator(func: Callable[_Param, Generator[_RetType, None, None]]) -> Callable[_Param, list[_RetType]]:
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs) -> list[_RetType]:
+        return list(func(*args, **kwargs))
+
+    return wrapper