test: fix tests

langflow-ai · Aug 28, 2024 · a27c6df · a27c6df
1 parent 2e560fa
commit a27c6df
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/backend/tests/unit/io/test_io_schema.py b/src/backend/tests/unit/io/test_io_schema.py
@@ -1,4 +1,4 @@
-from typing import List, Literal
+from typing import Literal
 
 import pytest
 from pydantic.fields import FieldInfo
@@ -96,7 +96,7 @@ def test_is_list_attribute_processing(self):
         input_instance = StrInput(name="test_field", is_list=True)
         schema = create_input_schema([input_instance])
         field_info: FieldInfo = schema.model_fields["test_field"]
-        assert field_info.annotation == List[str]
+        assert field_info.annotation == list[str]
 
     # Input with options attribute is processed correctly
     def test_options_attribute_processing(self):
@@ -184,7 +184,7 @@ def test_is_list_handling(self):
         input_instance = StrInput(name="test_field", is_list=True)
         schema = create_input_schema([input_instance])
         field_info = schema.model_fields["test_field"]
-        assert field_info.annotation == List[str]
+        assert field_info.annotation == list[str]
 
     # Converting FieldTypes to corresponding Python types
     def test_field_types_conversion(self):