feat(dask): implement json getitem

ibis-project · Sep 20, 2022 · 381d805 · 381d805
1 parent 9bfb748
commit 381d805
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ibis/backends/dask/execution/strings.py b/ibis/backends/dask/execution/strings.py
@@ -17,6 +17,8 @@
 )
 from ibis.backends.pandas.core import integer_types, scalar_types
 from ibis.backends.pandas.execution.strings import (
+    execute_json_getitem_series_series,
+    execute_json_getitem_series_str_int,
     execute_series_join_scalar_sep,
     execute_series_regex_extract,
     execute_series_regex_replace,
@@ -161,6 +163,10 @@
     ops.StringJoin: [
         (((dd.Series, str), list), execute_series_join_scalar_sep),
     ],
+    ops.JSONGetItem: [
+        ((dd.Series, (str, int)), execute_json_getitem_series_str_int),
+        ((dd.Series, dd.Series), execute_json_getitem_series_series),
+    ],
 }
 register_types_to_dispatcher(execute_node, DASK_DISPATCH_TYPES)