Update awkward.py

scikit-hep · Oct 17, 2023 · f6f3c75 · f6f3c75
1 parent 254e5c9
commit f6f3c75
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/vector/backends/awkward.py b/src/vector/backends/awkward.py
@@ -576,7 +576,10 @@ def elements(self) -> tuple[ArrayOrRecord]:
         return (self.tau,)
 
 
-def _class_to_name(cls: type[VectorProtocol]) -> str:
+T = typing.TypeVar("T", bound=VectorProtocol)
+
+
+def _class_to_name(cls: type[T]) -> str:
     if issubclass(cls, Momentum):
         if issubclass(cls, Vector2D):
             return "Momentum2D"