Use public cudf APIs where possible

rapidsai · Mar 6, 2024 · 839762e · 839762e
1 parent 02241e2
commit 839762e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/python/cuxfilter/charts/datashader/custom_extensions/graph_assets.py b/python/cuxfilter/charts/datashader/custom_extensions/graph_assets.py
@@ -82,7 +82,7 @@ def bundle_edges(edges, src="src", dst="dst"):
     # Determine each edge's index relative to its bundle
     edges = edges.sort_values(by="bid").reset_index(drop=True)
     edges["_index"] = (
-        cudf.core.index.RangeIndex(0, len(edges)) - edges["start"]
+        cudf.RangeIndex(0, len(edges)) - edges["start"]
     ).astype("int32")
 
     # Re-sort the edgelist by edge id and cleanup

diff --git a/python/cuxfilter/charts/datashader/custom_extensions/holoviews_datashader.py b/python/cuxfilter/charts/datashader/custom_extensions/holoviews_datashader.py
@@ -229,7 +229,7 @@ def __init__(self, **params):
     def _compute_clims(self):
         if not isinstance(
             self.source_df[self.aggregate_col].dtype,
-            cudf.core.dtypes.CategoricalDtype,
+            cudf.CategoricalDtype,
         ):
             self.clims = get_min_max(self.source_df, self.aggregate_col)
 
@@ -238,7 +238,7 @@ def _compute_datashader_assets(self):
         self.cmap = {"cmap": self.color_palette}
         if isinstance(
             self.source_df[self.aggregate_col].dtype,
-            cudf.core.dtypes.CategoricalDtype,
+            cudf.CategoricalDtype,
         ):
             self.cmap = {
                 "color_key": {