Use a more pythonic pattern

Gurobi · Nov 27, 2024 · c4225bb · c4225bb
1 parent 8c19d0e
commit c4225bb
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/src/gurobi_optimods/opf/grbformulator.py b/src/gurobi_optimods/opf/grbformulator.py
@@ -109,12 +109,8 @@ def lpformulator_optimize(alldata, model, opftype):
             bus = buses[j]
             evar[bus].PStart = 1.0
         if alldata["branchswitching_mip"]:
-            zvar = alldata["MIP"]["zvar"]
-            branches = alldata["branches"]
-            numbranches = alldata["numbranches"]
-            for j in range(1, 1 + numbranches):
-                branch = branches[j]
-                zvar[branch].PStart = 1.0
+            for zvar in alldata["MIP"]["zvar"].values():
+                zvar.PStart = 1.0
         model.update()
     model.optimize()