fix: typing import

scikit-hep · Jan 3, 2023 · 46ea4ec · 46ea4ec
1 parent 4b361bd
commit 46ea4ec
Showing 1 changed file with 16 additions and 10 deletions.
diff --git 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,