Force pydantic 2.0 (#42)

marda-alliance · Aug 22, 2023 · 7cdbc7d · 7cdbc7d
1 parent 45b3d43
commit 7cdbc7d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -3,5 +3,5 @@ invoke
 linkml
 mongomock
 pre-commit
-pydantic
+pydantic~=2.0
 uvicorn
diff --git a/tasks.py b/tasks.py
@@ -21,7 +21,7 @@ def regenerate_models(_):
     for schema in schemas:
         print(schema)
         schema_path = Path(schema)
-        gen = pd.PydanticGenerator(schema, verbose=True)
+        gen = pd.PydanticGenerator(schema, pydantic_version="2", verbose=True)
         output = gen.serialize()
         with open(MODEL_DIRECTORY / f"{schema_path.name.strip('.yml')}.py", "w") as f:
             f.writelines(output)