diff --git a/src/anemoi/models/models/encoder_processor_decoder.py b/src/anemoi/models/models/encoder_processor_decoder.py index aa7e8bb..c77db6e 100644 --- a/src/anemoi/models/models/encoder_processor_decoder.py +++ b/src/anemoi/models/models/encoder_processor_decoder.py @@ -108,7 +108,7 @@ def __init__( # Instantiation of model output bounding functions (e.g., to ensure outputs like TP are positive definite) self.boundings = nn.ModuleList( [ - instantiate(cfg, name_to_index=self.data_indices.model.output.name_to_index) + instantiate(cfg, name_to_index=self.data_indices.internal_model.output.name_to_index) for cfg in getattr(model_config.model, "bounding", []) ] )