Nit: Add a sort to the JSON schema generation for test consistency

huggingface · Jul 2, 2024 · 1446a25 · 1446a25
1 parent 0908989
commit 1446a25
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/utils/test_chat_template_utils.py b/tests/utils/test_chat_template_utils.py
@@ -137,7 +137,7 @@ def fn(x: List[List[Union[str, int]]]):
                 "properties": {
                     "x": {
                         "type": "array",
-                        "items": {"type": "array", "items": {"type": ["string", "integer"]}},
+                        "items": {"type": "array", "items": {"type": ["integer", "string"]}},
                         "description": "The input",
                     }
                 },
@@ -455,13 +455,13 @@ def fn(
                     },
                     "y": {
                         "type": "array",
-                        "items": {"type": ["string", "integer"]},
+                        "items": {"type": ["integer", "string"]},
                         "nullable": True,
                         "description": "The second input. It's a big list with a single-line description.",
                     },
                     "z": {
                         "type": "array",
-                        "prefixItems": [{"type": ["string", "integer"]}, {"type": "string"}],
+                        "prefixItems": [{"type": ["integer", "string"]}, {"type": "string"}],
                         "description": "The third input. It's some kind of tuple with a default arg.",
                     },
                 },