Use schema when generating tables

janheinrichmerker · Jul 3, 2024 · 5d6118e · 5d6118e
1 parent c17d413
commit 5d6118e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ray_elasticsearch/__init__.py b/ray_elasticsearch/__init__.py
@@ -130,7 +130,10 @@ def iter_blocks() -> Iterator[Table]:
                 hits = response["hits"]["hits"]
                 if len(hits) == 0:
                     break
-                yield Table.from_pylist(hits)
+                yield Table.from_pylist(
+                    mapping=hits,
+                    schema=schema,
+                )
                 search_after = max(
                     hit["sort"]
                     for hit in hits