Fixing schema

rosalindfranklininstitute · Jun 13, 2024 · 664d9b5 · 664d9b5
1 parent 36eca52
commit 664d9b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/ext/pydantic_settings.py b/docs/source/ext/pydantic_settings.py
@@ -85,7 +85,7 @@ def walk(obj):
 
         # Parse the definitions
         definitions = nodes.paragraph()
-        for key, value in schema["definitions"].items():
+        for key, value in schema["$defs"].items():
             definition = nodes.section(ids=[nodes.make_id(key)])
             definition += nodes.title(text=key)
             definition += walk(value)
@@ -108,7 +108,7 @@ def from_import(self, loc):
             raise e
 
         # Get the schema
-        return obj.schema()
+        return obj.model_json_schema()
 
 
 def setup(app):