From f6f3c75b347306f696c2e3db842f42c28371e33f Mon Sep 17 00:00:00 2001 From: Saswat Susmoy <72549122+Saswatsusmoy@users.noreply.github.com> Date: Wed, 18 Oct 2023 02:52:58 +0530 Subject: [PATCH] Update awkward.py --- src/vector/backends/awkward.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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"