[nfc] Remove/redirect some typing imports introduced in Python 3.11.

Author: stellaraccident

python/shark_turbine/kernel/_support/regions.py

@@ -3,7 +3,6 @@
     TypeVar,
     Callable,
     Type,
-    assert_type,
     cast,
     List,
     Dict,

python/shark_turbine/kernel/_support/tracing.py

@@ -4,7 +4,6 @@
     TypeVar,
     Callable,
     Type,
-    assert_type,
     cast,
     List,
     Dict,
@@ -42,6 +41,14 @@
 
 from . import context
 
+try:
+    from typing import assert_type
+except ImportError:
+    # No-op if not supported. Introduced in Python 3.11.
+    def assert_type(a, b):
+        pass
+
+
 TCallable = TypeVar("TCallable", bound=Callable)
 
 ###############################################################################

python/shark_turbine/kernel/gen/thread.py

@@ -5,7 +5,6 @@
     TypeVar,
     Callable,
     Union,
-    assert_type,
     cast,
     Any,
 )

python/shark_turbine/kernel/lang/prims.py

@@ -1,5 +1,3 @@
-from typing import assert_type
-
 from .. import ops
 
 from .._support.tracing import (

python/shark_turbine/kernel/ops/control_flow.py

@@ -7,7 +7,6 @@
     overload,
     Callable,
     Tuple,
-    Unpack,
 )
 import typing
 

python/shark_turbine/kernel/ops/memory.py

@@ -7,7 +7,6 @@
     overload,
     Callable,
     Tuple,
-    Unpack,
 )
 import typing