From 46ea4ec9116287ddd659e50094ae40ce5fbb8f45 Mon Sep 17 00:00:00 2001 From: Angus Hollands Date: Tue, 3 Jan 2023 21:05:05 +0000 Subject: [PATCH] fix: typing import --- src/awkward/typing.py | 26 ++++++++++++++++---------- 1 file changed, 16 insertions(+), 10 deletions(-) diff --git a/src/awkward/typing.py b/src/awkward/typing.py index 91afed70d2..36ad376b8b 100644 --- a/src/awkward/typing.py +++ b/src/awkward/typing.py @@ -5,16 +5,20 @@ import typing from typing import * # noqa: F403 -__all__ = [ - "Final", - "Self", - "final", - "Protocol", - "Unpack", - "TypeAlias", - "runtime_checkable", - "AxisMaybeNone", -] + typing.__all__ +__all__ = list( + { + "Final", + "Self", + "final", + "Protocol", + "Unpack", + "TypeAlias", + "runtime_checkable", + "AxisMaybeNone", + "TypedDict", + *typing.__all__, + } +) AxisMaybeNone = TypeVar("AxisMaybeNone", int, None) @@ -26,6 +30,7 @@ Protocol, Self, TypeAlias, + TypedDict, Unpack, final, runtime_checkable, @@ -36,6 +41,7 @@ Protocol, Self, TypeAlias, + TypedDict, Unpack, final, runtime_checkable,