restrict fix to typetracer backend

scikit-hep · Jan 21, 2025 · c05a481 · c05a481
1 parent 9c6a55a
commit c05a481
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/vector/backends/awkward.py b/src/vector/backends/awkward.py
@@ -649,7 +649,11 @@ class VectorAwkward:
 
     @property
     def lib(self):  # type:ignore[no-untyped-def]
-        return _lib(module=numpy, nplike=self.layout.backend.nplike)  # type:ignore[attr-defined]
+        if (
+            nplike := self.layout.backend.nplike  # type:ignore[attr-defined]
+        ) is ak._nplikes.typetracer.TypeTracer.instance():
+            return _lib(module=numpy, nplike=nplike)
+        return numpy
 
     def _wrap_result(
         self: AwkwardProtocol,