fix: list

instructor-ai · Apr 29, 2024 · aa0f429 · aa0f429
1 parent 503904a
commit aa0f429
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/test_simple_types.py b/tests/test_simple_types.py
@@ -1,4 +1,5 @@
-from typing import List
+from __future__ import annotations
+
 from instructor.dsl import is_simple_type, Partial
 from pydantic import BaseModel
 
@@ -62,6 +63,6 @@ def test_iterable_not_simple():
 
 
 def test_list_is_simple():
-    new_type = List[int]
+    new_type = list[int]
 
     assert is_simple_type(new_type), "Failed for type: " + str(new_type)