From 81eea77bc40c5188b2c5db04e7623f3c36749b5a Mon Sep 17 00:00:00 2001
From: Bryan Van de Ven <bryan@bokeh.org>
Date: Mon, 9 May 2022 12:32:22 -0700
Subject: [PATCH] Use stdlib typing where possible (#224)

---
 legate/core/constraints.py | 4 +---
 legate/core/launcher.py    | 3 ++-
 legate/core/operation.py   | 4 +---
 legate/core/shape.py       | 4 ++--
 legate/core/transform.py   | 3 +--
 5 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/legate/core/constraints.py b/legate/core/constraints.py
index 10ff3034a..ab961ecd9 100644
--- a/legate/core/constraints.py
+++ b/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
 
diff --git a/legate/core/launcher.py b/legate/core/launcher.py
index cfe800697..51a2269c2 100644
--- a/legate/core/launcher.py
+++ b/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,
diff --git a/legate/core/operation.py b/legate/core/operation.py
index b47c2f368..05934a2da 100644
--- a/legate/core/operation.py
+++ b/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
 
diff --git a/legate/core/shape.py b/legate/core/shape.py
index 67f69e56a..371c6b388 100644
--- a/legate/core/shape.py
+++ b/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
diff --git a/legate/core/transform.py b/legate/core/transform.py
index c120132b3..d4fa38b42 100644
--- a/legate/core/transform.py
+++ b/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