diff --git a/src/vector/backends/awkward.py b/src/vector/backends/awkward.py index ed6a0974..d8906e6a 100644 --- 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"