if not split documents then return after yield

MeltanoLabs · Aug 7, 2023 · eb0a2d3 · eb0a2d3
1 parent 03591dd
commit eb0a2d3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/map_gpt_embeddings/mappers.py b/map_gpt_embeddings/mappers.py
@@ -106,6 +106,7 @@ def split_record(self, record: dict) -> t.Iterable[dict]:
 
         if not self.config.get("split_documents", True):
             yield record
+            return
 
         splitter_config = self.config.get("splitter_config", {})
         if "chunk_size" not in splitter_config: