typing

dask-contrib · Oct 31, 2023 · 0d2939f · 0d2939f
1 parent c7f5fbb
commit 0d2939f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dask_awkward/layers/layers.py b/src/dask_awkward/layers/layers.py
@@ -93,7 +93,7 @@ def mock(self) -> AwkwardArray:
         assert self._meta is not None
         return self._meta
 
-    def mock_empty(self, backend: str = "cpu") -> AwkwardArray:
+    def mock_empty(self, backend: BackendT = "cpu") -> AwkwardArray:
         import awkward as ak
 
         if backend not in ("cpu", "jax", "cuda"):