⬆️ Upgrade to pydantic v2

laminlabs · Jul 23, 2024 · 5080aaf · 5080aaf
1 parent ea86a49
commit 5080aaf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/nbproject/dev/_meta_store.py b/nbproject/dev/_meta_store.py
@@ -1,7 +1,7 @@
 from pathlib import Path
 from typing import List, Mapping, Optional, Union
 
-from pydantic import BaseModel, Extra
+from pydantic import BaseModel, ConfigDict
 
 from nbproject._logger import logger
 
@@ -61,9 +61,7 @@ class MetaContainer(BaseModel):
     """User ID from lamindb."""
     user_name: Optional[str] = None
     """User name from lamindb."""
-
-    class Config:
-        extra = Extra.allow
+    model_config = ConfigDict(extra="allow")
 
 
 class MetaStore:

diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ classifiers = [
     "Programming Language :: Python :: 3.11",
 ]
 dependencies = [
-    "pydantic<2.0.0",
+    "pydantic>=2.0.0",
     "pyyaml",
     "packaging",
     "orjson",