Use stdlib typing where possible (nv-legate#224)

legate/core/constraints.py

@@ -15,9 +15,7 @@
 from __future__ import annotations
 
 from collections.abc import Iterable
-from typing import TYPE_CHECKING, Any, Iterator, Optional, Union
-
-from typing_extensions import Protocol
+from typing import TYPE_CHECKING, Any, Iterator, Optional, Protocol, Union
 
 from .partition import Restriction
 

legate/core/launcher.py

@@ -20,13 +20,14 @@
     Any,
     Callable,
     Optional,
+    Protocol,
     Sequence,
     Tuple,
     Union,
+    overload,
 )
 
 import pyarrow as pa
-from typing_extensions import Protocol, overload
 
 from . import (
     ArgumentMap,

legate/core/operation.py

@@ -14,9 +14,7 @@
 #
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Iterable, Optional, Union
-
-from typing_extensions import Protocol
+from typing import TYPE_CHECKING, Any, Iterable, Optional, Protocol, Union
 
 import legate.core.types as ty
 

legate/core/shape.py

@@ -15,9 +15,9 @@
 from __future__ import annotations
 
 from functools import reduce
-from typing import TYPE_CHECKING, Iterable, Iterator, Optional, Union
+from typing import TYPE_CHECKING, Iterable, Iterator, Optional, Union, overload
 
-from typing_extensions import TypeAlias, overload
+from typing_extensions import TypeAlias
 
 if TYPE_CHECKING:
     from . import IndexSpace

legate/core/transform.py

@@ -14,10 +14,9 @@
 #
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Tuple
+from typing import TYPE_CHECKING, Protocol, Tuple
 
 import numpy as np
-from typing_extensions import Protocol
 
 from . import AffineTransform
 from .partition import Replicate, Restriction, Tiling