Fix mypy type stub packages in pre-commit and pydanticgen (#17)

* Fix mypy type stub packages in pre-commit * Ensure pydantic v2 is installed * Try different version specifier * Remove now-defunct parameter in pydantic generation
datatractor · Sep 10, 2024 · cd9af9a · cd9af9a
1 parent 6ff7e6c
commit cd9af9a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -39,4 +39,4 @@ repos:
       hooks:
           - id: mypy
             name: MyPy
-            additional_dependencies: [types-all, pydantic~=1.10]
+            additional_dependencies: [types-PyYAML, pydantic>=2]
diff --git a/tasks.py b/tasks.py
@@ -23,7 +23,7 @@ def regenerate_models(_):
     for schema in schemas:
         print(schema)
         schema_path = Path(schema)
-        gen = pd.PydanticGenerator(schema, pydantic_version="2", verbose=True)
+        gen = pd.PydanticGenerator(schema, verbose=True)
         output = gen.serialize()
         with open(MODEL_DIRECTORY / f"{schema_path.name.strip('.yaml')}.py", "w") as f:
             f.writelines(output)