diff --git a/antarest/study/business/arrow_utils.py b/antarest/study/business/arrow_utils.py index 8f23a7b232..b08e9531f5 100644 --- a/antarest/study/business/arrow_utils.py +++ b/antarest/study/business/arrow_utils.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024, RTE (https://www.rte-france.com) +# Copyright (c) 2025, RTE (https://www.rte-france.com) # # See AUTHORS.txt # diff --git a/antarest/study/business/load_management.py b/antarest/study/business/load_management.py index fce3d33a23..00d889eb89 100644 --- a/antarest/study/business/load_management.py +++ b/antarest/study/business/load_management.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024, RTE (https://www.rte-france.com) +# Copyright (c) 2025, RTE (https://www.rte-france.com) # # See AUTHORS.txt # @@ -36,9 +36,8 @@ def get_load_matrix(self, study: Study, area_id: str) -> t.Tuple[pd.DataFrame, t if not isinstance(node, InputSeriesMatrix): raise TypeError(f"Expected node of type 'InputSeriesMatrix', but got '{type(node).__name__}'") - matrix_data = InputSeriesMatrix.parse(node, return_dataframe=True) + matrix_df = InputSeriesMatrix.parse_as_dataframe(node) - matrix_df = t.cast(pd.DataFrame, matrix_data) matrix_df.columns = matrix_df.columns.map(str) matrix_df.columns = pd.Index(matrix_columns) diff --git a/tests/integration/study_data_blueprint/test_load.py b/tests/integration/study_data_blueprint/test_load.py index 67e5f88fb2..5af3a8ee13 100644 --- a/tests/integration/study_data_blueprint/test_load.py +++ b/tests/integration/study_data_blueprint/test_load.py @@ -1,4 +1,4 @@ -# Copyright (c) 2024, RTE (https://www.rte-france.com) +# Copyright (c) 2025, RTE (https://www.rte-france.com) # # See AUTHORS.txt #