add from_legacy_dataframe calls

lincc-frameworks · Apr 26, 2024 · aa2c0a0 · aa2c0a0
1 parent 6c8f1b4
commit aa2c0a0
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py
@@ -6,6 +6,7 @@
 import requests
 import lsdb
 import dask
+from dask_expr import from_legacy_dataframe
 
 import dask.dataframe as dd
 import numpy as np
@@ -1721,7 +1722,7 @@ def from_lsdb(
                     sort = True
 
             self.from_dask_dataframe(
-                source_catalog._ddf,
+                from_legacy_dataframe(source_catalog._ddf),
                 None,
                 column_mapper=column_mapper,
                 sync_tables=sync_tables,
@@ -1744,8 +1745,8 @@ def from_lsdb(
                 sort = True
 
             self.from_dask_dataframe(
-                source_catalog._ddf,
-                object_catalog._ddf,
+                from_legacy_dataframe(source_catalog._ddf),
+                 from_legacy_dataframe(object_catalog._ddf),
                 column_mapper=column_mapper,
                 sync_tables=sync_tables,
                 sorted=sorted,