From 7e0671a634107744c88b490ddfa40295a4574666 Mon Sep 17 00:00:00 2001 From: larsevj Date: Tue, 8 Oct 2024 11:05:32 +0200 Subject: [PATCH] Try correct typing --- src/ert/config/design_matrix.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ert/config/design_matrix.py b/src/ert/config/design_matrix.py index c05790a3ffb..641233631d5 100644 --- a/src/ert/config/design_matrix.py +++ b/src/ert/config/design_matrix.py @@ -28,7 +28,7 @@ class DesignMatrix: design_sheet: str default_sheet: str design_matrix_df: Optional[pd.DataFrame] = None - parameter_configuration: Optional[list[ParameterConfig]] = None + parameter_configuration: Optional[dict[str, ParameterConfig]] = None @classmethod def from_config_list(cls, config_list: List[str]) -> "DesignMatrix": @@ -97,7 +97,7 @@ def read_design_matrix( # ignoring errors here is deprecated in pandas, should find another solution # design_matrix_sheet = design_matrix_sheet.apply(pd.to_numeric, errors="ignore") - parameter_configuration = {} + parameter_configuration: dict[str, ParameterConfig] = {} transform_function_definitions: list[TransformFunctionDefinition] = [] for parameter in design_matrix_df.columns: transform_function_definitions.append(