more fixes

rapidsai · Feb 9, 2024 · 4fd9ea6 · 4fd9ea6
1 parent 7b5b5c0
commit 4fd9ea6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/python/cuxfilter/charts/core/aggregate/core_aggregate.py b/python/cuxfilter/charts/core/aggregate/core_aggregate.py
@@ -187,12 +187,12 @@ def reset_callback(resetting):
     def get_box_select_callback(self, dashboard_cls):
         def cb(bounds, x_selection, y_selection):
             self.box_selected_range, self.selected_indices = None, None
-            if type(x_selection) == tuple:
+            if isinstance(x_selection, tuple):
                 self.box_selected_range = {
                     self.x + "_min": x_selection[0],
                     self.x + "_max": x_selection[1],
                 }
-            elif type(x_selection) == list:
+            elif isinstance(x_selection, list):
                 self.selected_indices = (
                     dashboard_cls._cuxfilter_df.data[self.x]
                     .isin(x_selection)

diff --git a/python/cuxfilter/dashboard.py b/python/cuxfilter/dashboard.py
@@ -136,6 +136,8 @@ def queried_indices(self):
             for (key, value) in self._query_str_dict.items()
             if type(value)
             in [
+                cudf.DataFrame,
+                dask_cudf.DataFrame,
                 cudf.Series,
                 dask_cudf.Series,
             ]