Quick test fix

huggingface · May 9, 2024 · 295b89d · 295b89d
1 parent 3d8c43f
commit 295b89d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/utils/test_chat_template_utils.py b/tests/utils/test_chat_template_utils.py
@@ -101,16 +101,14 @@ def fn(x: int = 42):
         self.assertEqual(schema, expected_schema)
 
     def test_nested_list(self):
-        def fn(x: List[List[Union[int, str]]]):
+        def fn(x: List[List[Union[str, int]]]):
             """
             Test function
 
             :param x: The input
             """
             return x
 
-        self.maxDiff = None
-
         schema = get_json_schema(fn)
         expected_schema = {
             "name": "fn",
@@ -120,7 +118,7 @@ def fn(x: List[List[Union[int, str]]]):
                 "properties": {
                     "x": {
                         "type": "array",
-                        "items": {"type": "array", "items": {"type": ["integer", "string"]}},
+                        "items": {"type": "array", "items": {"type": ["string", "integer"]}},
                         "description": "The input",
                     }
                 },