diff --git a/src/tape/ensemble.py b/src/tape/ensemble.py index 9a6ef7f9..140ed52e 100644 --- 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,