python_generator_test: Remove usage of typing_extensions

All the used methods are available in the native typing module since python 3.8
google · May 17, 2024 · 979c751 · 979c751
1 parent fe1935b
commit 979c751
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/pdl-compiler/tests/python_generator_test.py b/pdl-compiler/tests/python_generator_test.py
@@ -21,7 +21,6 @@
 import enum
 import json
 import typing
-import typing_extensions
 import unittest
 from importlib import resources
 
@@ -57,12 +56,12 @@ def create_object(typ, value):
             field_type = field_types[f]
             values[f] = create_object(field_type, v)
         return typ(**values)
-    elif typing_extensions.get_origin(typ) is list:
-        typ = typing_extensions.get_args(typ)[0]
+    elif typing.get_origin(typ) is list:
+        typ = typing.get_args(typ)[0]
         return [create_object(typ, v) for v in value]
-    elif typing_extensions.get_origin(typ) is typing.Union:
+    elif typing.get_origin(typ) is typing.Union:
         # typing.Optional[int] expands to typing.Union[int, None]
-        typ = typing_extensions.get_args(typ)[0]
+        typ = typing.get_args(typ)[0]
         return create_object(typ, value) if value is not None else None
     elif typ is bytes:
         return bytes(value)