diff --git a/pyproject.toml b/pyproject.toml index 67896aec..a93e58c9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -43,7 +43,8 @@ dependencies = [ "prettytable==3.13.*", "pydantic>=2.0.0", "typeguard==2.13.3", - "ydata-datascience" + "ydata-datascience", + "requests==2.*", ] [project.license] diff --git a/src/ydata/sdk/common/status.py b/src/ydata/sdk/common/status.py index 8427adac..5e14dece 100644 --- a/src/ydata/sdk/common/status.py +++ b/src/ydata/sdk/common/status.py @@ -1,6 +1,6 @@ from typing import Generic, Optional, TypeVar -from pydantic import Field +from pydantic import ConfigDict, Field from .model import BaseModel @@ -8,7 +8,7 @@ class GenericStateErrorStatus(BaseModel, Generic[T]): + model_config = ConfigDict( + use_enum_values=True + ) state: Optional[T] = Field(None) - - class Config: - use_enum_values = True diff --git a/src/ydata/sdk/datasources/_models/metadata/column.py b/src/ydata/sdk/datasources/_models/metadata/column.py index 1cc50275..ebf816e9 100644 --- a/src/ydata/sdk/datasources/_models/metadata/column.py +++ b/src/ydata/sdk/datasources/_models/metadata/column.py @@ -1,18 +1,18 @@ -from pydantic import Field +from pydantic import ConfigDict, Field from ydata.sdk.common.model import BaseModel from ydata.sdk.datasources._models.metadata.data_types import DataType, VariableType class Column(BaseModel): + model_config = ConfigDict( + use_enum_values=True + ) name: str datatype: DataType = Field(alias='dataType') vartype: VariableType = Field(alias='varType') - class Config: - use_enum_values = True - def __repr__(self) -> str: return f"Column(name={self.name}, datatype={self.datatype}, vartype={self.vartype})"