support ak.str called on dak collections

dask-contrib · Sep 6, 2023 · 653eab2 · 653eab2
1 parent e1aa0c3
commit 653eab2
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/dask_awkward/lib/core.py b/src/dask_awkward/lib/core.py
@@ -1273,7 +1273,12 @@ def __awkward_function__(self, func, array_likes, args, kwargs):
         try:
             fn = getattr(dask_awkward, fn_name)
         except AttributeError:
-            return NotImplemented
+            try:
+                import dask_awkward.lib.str
+
+                fn = getattr(dask_awkward.str, fn_name)
+            except AttributeError:
+                return NotImplemented
         return fn(*args, **kwargs)
 
     def __array_ufunc__(self, ufunc, method, *inputs, **kwargs):