fix tests

AntaresSimulatorTeam · Sep 25, 2024 · fe17ecf · fe17ecf
1 parent 29c58c7
commit fe17ecf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/antarest/study/model.py b/antarest/study/model.py
@@ -19,7 +19,7 @@
 from pathlib import Path
 
 from antares.study.version import StudyVersion
-from pydantic import BaseModel, field_validator, field_serializer
+from pydantic import BaseModel, field_serializer, field_validator
 from sqlalchemy import (  # type: ignore
     Boolean,
     Column,
@@ -359,7 +359,7 @@ class StudyMetadataDTO(BaseModel):
     folder: t.Optional[str] = None
     tags: t.List[str] = []
 
-    @field_serializer('version')
+    @field_serializer("version")
     def serialize_version(self, version: StudyVersion) -> int:
         return version.__int__()
 

diff --git a/antarest/study/storage/rawstudy/model/filesystem/config/model.py b/antarest/study/storage/rawstudy/model/filesystem/config/model.py
@@ -15,7 +15,7 @@
 from pathlib import Path
 
 from antares.study.version import StudyVersion
-from pydantic import BaseModel, Field, model_validator, field_serializer, field_validator
+from pydantic import BaseModel, Field, field_serializer, field_validator, model_validator
 
 from antarest.core.utils.utils import DTO
 from antarest.study.business.enum_ignore_case import EnumIgnoreCase
@@ -318,15 +318,14 @@ class FileStudyTreeConfigDTO(BaseModel):
     enr_modelling: str = str(EnrModelling.AGGREGATED)
     zip_path: t.Optional[Path] = None
 
-    @field_serializer('version')
+    @field_serializer("version")
     def serialize_version(self, version: StudyVersion) -> int:
         return version.__int__()
 
     @field_validator("version", mode="before")
     def _validate_version(cls, v: t.Any) -> StudyVersion:
         return StudyVersion.parse(v)
 
-
     @staticmethod
     def from_build_config(
         config: FileStudyTreeConfig,