From 82e6c8dd337ea7b0f65d3ba19b24f91c20161796 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 767e3e0d..ef694dc9 100644 --- a/src/vector/backends/awkward.py +++ b/src/vector/backends/awkward.py @@ -584,7 +584,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: # respect the type of classes inheriting VectorAwkward classes is_vector = "vector.backends" in cls.__module__ if issubclass(cls, Momentum):