Lock jsonschema version

Kiln-AI · Sep 9, 2024 · 737aba6 · 737aba6
1 parent cb4d756
commit 737aba6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/libs/core/kiln_ai/datamodel/json_schema.py b/libs/core/kiln_ai/datamodel/json_schema.py
@@ -3,6 +3,7 @@
 
 import jsonschema
 import jsonschema.exceptions
+import jsonschema.validators
 from pydantic import AfterValidator
 
 JsonObjectSchema = Annotated[
@@ -19,7 +20,8 @@ def _check_json_schema(v: str) -> str:
 
 def validate_schema(instance: Dict, schema_str: str) -> None:
     schema = schema_from_json_str(schema_str)
-    jsonschema.validate(instance=instance, schema=schema)
+    v = jsonschema.Draft202012Validator(schema)
+    return v.validate(instance)
 
 
 def schema_from_json_str(v: str) -> Dict: