test fix

Author: Zulko

test/test_gpt_function.py

@@ -1,5 +1,6 @@
 from gpt_function_decorator import gpt_function
 from pydantic import BaseModel, Field
+from typing import List
 
 
 def test_interpolated_docstring():
@@ -28,7 +29,7 @@ def positivity(sentence) -> int:
 
 def test_list_of_str_output():
     @gpt_function
-    def list_famous_composers(n) -> list[str]:
+    def list_famous_composers(n) -> List[str]:
         "Return the {n} most famous composers."
 
     assert len(list_famous_composers(2)) == 2
@@ -40,7 +41,7 @@ class USPresident(BaseModel):
         name: str
 
     @gpt_function
-    def first_us_presidents(n) -> list[USPresident]:
+    def first_us_presidents(n) -> List[USPresident]:
         """Return the {n} first US presidents with their birth year"""
 
     assert first_us_presidents(1)[0].birth_year == 1732
@@ -52,7 +53,7 @@ class USPresident(BaseModel):
         name: str = Field(description="Family name")
 
     @gpt_function
-    def first_us_presidents(n) -> list[USPresident]:
+    def first_us_presidents(n) -> List[USPresident]:
         """Return the {n} first US presidents with their birth year"""
 
     assert first_us_presidents(1)[0].name == "Washington"
@@ -73,7 +74,7 @@ def from_description(description) -> "Car":
 
 def test_reasoned_answer():
     @gpt_function
-    def could_have_met(person: str, celebrities: list) -> list[str]:
+    def could_have_met(person: str, celebrities: list) -> List[str]:
         """List the names in {celebrities} that {person} could have met."""
 
     answer = could_have_met(