CU-8694dpy1c: Return empty generator upon empty stream (#423)

* CU-8694dpy1c: Return empty generator upon empty stream * CU-8694dpy1c: Fix empty generator returns * CU-8694dpy1c: Simplify empty generator returns
CogStack · Aug 13, 2024 · a51b35f · a51b35f
1 parent 75ca4e2
commit a51b35f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/medcat/meta_cat.py b/medcat/meta_cat.py
@@ -481,7 +481,8 @@ def pipe(self, stream: Iterable[Union[Doc, FakeDoc]], *args, **kwargs) -> Iterat
         """
         # Just in case
         if stream is None or not stream:
-            return stream
+            # return an empty generator
+            return
 
         config = self.config
         id2category_value = {v: k for k, v in config.general['category_value2id'].items()}

diff --git a/medcat/ner/transformers_ner.py b/medcat/ner/transformers_ner.py
@@ -359,7 +359,8 @@ def pipe(self, stream: Iterable[Union[Doc, None]], *args, **kwargs) -> Iterator[
         """
         # Just in case
         if stream is None or not stream:
-            return stream
+            # return an empty generator
+            return
 
         batch_size_chars = self.config.general['pipe_batch_size_in_chars']
         yield from self._process(stream, batch_size_chars)  # type: ignore