feat: support ak.str as dak.str

src/dask_awkward/__init__.py

@@ -6,6 +6,7 @@
 import dask_awkward.lib.operations as operations
 import dask_awkward.lib.optimize as optimize
 import dask_awkward.lib.reducers as reducers
+import dask_awkward.lib.str as str
 import dask_awkward.lib.structure as structure
 from dask_awkward.lib.core import Array, PartitionCompatibility, Record, Scalar
 from dask_awkward.lib.core import _type as type

src/dask_awkward/lib/__init__.py

@@ -1,3 +1,4 @@
+import dask_awkward.lib.str as str
 from dask_awkward.lib.core import Array, PartitionCompatibility, Record, Scalar
 from dask_awkward.lib.core import _type as type
 from dask_awkward.lib.core import (

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):