fix: failed imports from typing_extensions 4.5.0 in Colab (#1620)

* fix imports for typing_extensions 4.5.0 * put version ceiling on typing_extensions for CI
Arize-ai · Oct 13, 2023 · f57c4bb · f57c4bb
1 parent c092387
commit f57c4bb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -112,7 +112,8 @@ dependencies = [
   "requests",
   "protobuf==3.20",  # version minimum (for tests)
   "responses",
-  "tiktoken"
+  "tiktoken",
+  "typing-extensions<4.6.0",  # for Colab
 ]
 
 [tool.hatch.envs.type]

diff --git a/src/phoenix/utilities/error_handling.py b/src/phoenix/utilities/error_handling.py
@@ -1,8 +1,6 @@
 import logging
 import traceback
-from typing import Any, Callable, Iterable, Optional, Type
-
-from typing_extensions import TypeVar, cast
+from typing import Any, Callable, Iterable, Optional, Type, TypeVar, cast
 
 F = TypeVar("F", bound=Callable[..., Any])