Fix _update_variable_names!

spine-tools · Oct 24, 2023 · bad15aa · bad15aa
1 parent 8c4267f
commit bad15aa
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/run_spineopt_standard.jl b/src/run_spineopt_standard.jl
@@ -933,8 +933,10 @@ end
 
 function _update_variable_names!(m)
     for (name, var) in m.ext[:spineopt].variables
-        for (ind, v) in var
-            _set_name(v, _base_name(name, ind))
+        # NOTE: only update names for the representative variables
+        # This is achieved by using the indices function from the variable definition
+        for ind in m.ext[:spineopt].variables_definition[name][:indices](m)
+            _set_name(var[ind], _base_name(name, ind))
         end
     end
 end