mypy

Author: gabriel

dataframely/functional.py

@@ -107,7 +107,7 @@ def concat_collection_members(collections: Sequence[C], /) -> dict[str, pl.LazyF
     """
     if len(collections) == 0:
         raise ValueError("Cannot concatenate less than one collection.")
-    members = [c.to_dict() for c in collections]
+    members = [c._to_lazy_dict() for c in collections]
     key_union = set(members[0]).union(*members[1:])
     return {
         key: pl.concat(

dataframely/testing/storage.py

@@ -276,7 +276,7 @@ def write_untyped(
         # Since the metadata retrieval depends on the metadata being attached to the
         # latest commit, this will make the metadata unretrievable.
         fs: AbstractFileSystem = url_to_fs(path)[0]
-        for member, df in collection.to_dict().items():
+        for member, df in collection._to_lazy_dict().items():
             table = _to_delta_table(fs.sep.join([path, member]))
             df.head(0).collect().write_delta(table, mode="append")