From aa2c0a0b23952c65e2405224c7f3f20142a4022e Mon Sep 17 00:00:00 2001 From: Doug Branton Date: Fri, 26 Apr 2024 13:26:55 -0700 Subject: [PATCH] add from_legacy_dataframe calls --- src/tape/ensemble.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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,