diff --git a/pyproject.toml b/pyproject.toml index a4b5e061..67896aec 100644 --- 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 index cb286263..bd11b14a 100644 --- 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 index e8696e91..5a8768ef 100644 --- 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