Merge pull request #438 from lincc-frameworks/fix_lsdb_compat

add from_legacy_dataframe calls
lincc-frameworks · Apr 26, 2024 · 40236a3 · 40236a3
2 parents 6c8f1b4 + 3e05011
commit 40236a3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/docs/tutorials/working_with_hipscat_and_lsdb.ipynb b/docs/tutorials/working_with_hipscat_and_lsdb.ipynb
@@ -175,7 +175,7 @@
     "object_cat_cone = object_cat.cone_search(\n",
     "    ra=315.0,\n",
     "    dec=-69.5,\n",
-    "    radius_arcsec=10.0,\n",
+    "    radius_arcsec=100000.0,\n",
     ")\n",
     "\n",
     "print(f\"Original Number of Objects: {len(object_cat._ddf)}\")\n",

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,