refactoring

godatadriven · Mar 4, 2024 · d8eb7f1 · d8eb7f1
1 parent d832d7b
commit d8eb7f1
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/pydantic_spark/base.py b/src/pydantic_spark/base.py
@@ -89,7 +89,7 @@ def get_type(value: dict) -> Tuple[str, dict]:
             if "default" in value:
                 metadata["default"] = value.get("default")
             if r is not None:
-                class_name = r.replace("#/definitions/", "")
+                class_name = r.replace("#/$defs/", "")
                 if class_name in classes_seen:
                     spark_type = classes_seen[class_name]
                 else:
@@ -138,7 +138,6 @@ def get_type(value: dict) -> Tuple[str, dict]:
                     f"Type '{t}' not support yet, "
                     f"please report this at https://github.com/godatadriven/pydantic-avro/issues"
                 )
-            print(spark_type)
             return spark_type, metadata
 
         def get_fields(s: dict) -> List[dict]: