fix: pydantic and typeguard (#153)

* fix: pydantic and typeguard * fix(linting): code formatting --------- Co-authored-by: Azory YData Bot <[email protected]>
ydataai · Feb 3, 2025 · c6c7277 · c6c7277
1 parent 6b56714
commit c6c7277
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 14 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -42,7 +42,7 @@ dependencies = [
     "pandas>=1.5.0",
     "prettytable==3.13.*",
     "pydantic>=2.0.0",
-    "typeguard>=4.0.0",
+    "typeguard==2.13.3",
     "ydata-datascience"
 ]
 

diff --git a/src/ydata/sdk/common/model.py b/src/ydata/sdk/common/model.py
@@ -1,10 +1,10 @@
 from pydantic import BaseModel as PydanticBaseModel
-from pydantic import Extra
+from pydantic import ConfigDict
 
 
-class Config:
-    allow_population_by_field_name = True
-    extra = Extra.ignore
+class Config(ConfigDict):
+    populate_by_name = True
+    extra = 'allow'
     use_enum_values = True
 
 
@@ -13,4 +13,4 @@ class BaseModel(PydanticBaseModel):
 
     All datamodel from YData SDK inherits from this class.
     """
-    Config = Config
+    model_config = Config()
diff --git a/src/ydata/sdk/connectors/_models/schema.py b/src/ydata/sdk/connectors/_models/schema.py
@@ -2,16 +2,17 @@
 
 from pydantic import Field
 
-from ydata.sdk.common.model import BaseModel
+from ydata.sdk.common.model import BaseModel, Config
 from ydata.sdk.common.pydantic_utils import to_camel
 
 
-class BaseConfig(BaseModel.Config):
+class BaseConfig(Config):
     alias_generator = to_camel
 
 
 class TableColumn(BaseModel):
     """Class to store the information of a Column table."""
+    model_config = BaseConfig()
 
     name: str
     variable_type: str  # change this to the datatypes
@@ -20,24 +21,20 @@ class TableColumn(BaseModel):
     foreign_keys: list
     nullable: bool
 
-    Config = BaseConfig
-
 
 class Table(BaseModel):
     """Class to store the table columns information."""
+    model_config = BaseConfig()
 
     name: str
     columns: List[TableColumn]
     primary_keys: List[TableColumn]
     foreign_keys: List[TableColumn]
 
-    Config = BaseConfig
-
 
 class Schema(BaseModel):
     """Class to store the database schema information."""
+    model_config = BaseConfig()
 
     name: str
     tables: Optional[List[Table]] = Field(None)
-
-    Config = BaseConfig