fix test

AntaresSimulatorTeam · Oct 1, 2024 · 6271165 · 6271165
1 parent 534df03
commit 6271165
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/antarest/study/storage/rawstudy/model/filesystem/matrix/date_serializer.py b/antarest/study/storage/rawstudy/model/filesystem/matrix/date_serializer.py
@@ -110,7 +110,7 @@ def extract_date(self, df: pd.DataFrame) -> Tuple[pd.Index[str], pd.DataFrame]:
         to_remove = cast(Sequence[Hashable], df.columns[0:5])
         body = df.drop(to_remove, axis=1)
 
-        return cast(pd.Index[str], pd.Index(date)), body
+        return pd.Index(date), body  # type: ignore
 
 
 class DailyMatrixSerializer(IDateMatrixSerializer):
@@ -148,7 +148,7 @@ def extract_date(self, df: pd.DataFrame) -> Tuple[pd.Index[str], pd.DataFrame]:
         to_remove = cast(Sequence[Hashable], df.columns[0:4])
         body = df.drop(to_remove, axis=1)
 
-        return cast(pd.Index[str], pd.Index(date)), body
+        return pd.Index(date), body  # type: ignore
 
 
 class WeeklyMatrixSerializer(IDateMatrixSerializer):