chore: add type hints

dask-contrib · Oct 3, 2023 · 379400d · 379400d
1 parent 8e54d0b
commit 379400d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/dask_awkward/lib/inspect.py b/src/dask_awkward/lib/inspect.py
@@ -5,10 +5,12 @@
 import numpy as np
 
 if TYPE_CHECKING:
+    from awkward.highlevel import Array as AwkArray
+
     from dask_awkward.lib.core import Array
 
 
-def _random_boolean_like(array_like, probability):
+def _random_boolean_like(array_like: AwkArray, probability: float) -> AwkArray:
     import awkward as ak
 
     backend = ak.backend(array_like)