Handle if there are more states to be initialized then length of vari…

…able
dnv-opensource · Mar 12, 2024 · e521093 · e521093
1 parent 30deb57
commit e521093
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/mlfmu/types/fmu_component.py b/src/mlfmu/types/fmu_component.py
@@ -456,6 +456,8 @@ def get_template_mapping(
                 input_mapping.append((input_index, inp.variable_references[variable_index]))
 
             for variable_index, state_init_index in enumerate(inp.agent_state_init_indexes):
+                if variable_index >= len(inp.variable_references):
+                    break
                 state_init_mapping.append((state_init_index, inp.variable_references[variable_index]))
 
         for out in self.outputs: