Fix NoneType type as it requires py>=3.10 (#35843)

fix type
huggingface · Jan 30, 2025 · b17abf9 · b17abf9
1 parent f7b6047
commit b17abf9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/transformers/utils/chat_template_utils.py b/src/transformers/utils/chat_template_utils.py
@@ -77,6 +77,7 @@ def _get_json_schema_type(param_type: str) -> Dict[str, str]:
         float: {"type": "number"},
         str: {"type": "string"},
         bool: {"type": "boolean"},
+        type(None): {"type": "null"},
         Any: {},
     }
     if is_vision_available():