Quick test fixes

huggingface · May 24, 2024 · f6fcf04 · f6fcf04
1 parent 831262b
commit f6fcf04
Showing 1 changed file with 4 additions and 4 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": ["integer", "string"]}},
+                        "items": {"type": "array", "items": {"type": ["string", "integer"]}},
                         "description": "The input",
                     }
                 },
@@ -418,7 +418,7 @@ def fn(x: int, y: int):
 
     def test_everything_all_at_once(self):
         def fn(
-            x: str, y: Optional[List[Union[int, str]]], z: Tuple[Union[int, str], str] = (42, "hello")
+            x: str, y: Optional[List[Union[str, int]]], z: Tuple[Union[str, int], str] = (42, "hello")
         ) -> Tuple[int, str]:
             """
             Test function with multiple args, and docstring args that we have to strip out.
@@ -448,13 +448,13 @@ def fn(
                     "x": {"type": "string", "description": "The first input. It's got a big multiline"},
                     "y": {
                         "type": "array",
-                        "items": {"type": ["integer", "string"]},
+                        "items": {"type": ["string", "integer"]},
                         "nullable": True,
                         "description": "The second input. It's a big list with a single-line description.",
                     },
                     "z": {
                         "type": "array",
-                        "prefixItems": [{"type": ["integer", "string"]}, {"type": "string"}],
+                        "prefixItems": [{"type": ["string", "integer"]}, {"type": "string"}],
                         "description": "The third input. It's some kind of tuple with a default arg.",
                     },
                 },