feat(ingest/jsonschema) enable schema-aware JsonSchemaTranslator (#…

…8971) Co-authored-by: Harshal Sheth <[email protected]>
datahub-project · Oct 30, 2023 · 758ed47 · 758ed47
1 parent 3f4ab44
commit 758ed47
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/metadata-ingestion/src/datahub/ingestion/extractor/json_schema_util.py b/metadata-ingestion/src/datahub/ingestion/extractor/json_schema_util.py
@@ -598,7 +598,8 @@ def get_fields_from_schema(
                     jsonref_schema_dict = schema_dict
                 else:
                     # first validate the schema using a json validator
-                    jsonschema.Draft7Validator.check_schema(schema_dict)
+                    validator = jsonschema.validators.validator_for(schema_dict)
+                    validator.check_schema(schema_dict)
                     # then apply jsonref
                     jsonref_schema_dict = jsonref.loads(schema_string)
             except Exception as e: