fix: correct bug when MW-mesh scaling with existing ct

Breakthrough-Energy · Apr 7, 2020 · 32ff8b4 · 32ff8b4
1 parent 6826b6a
commit 32ff8b4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/powersimdata/input/design_transmission.py b/powersimdata/input/design_transmission.py
@@ -211,7 +211,8 @@ def _identify_mesh_branch_upgrades(ref_scenario, upgrade_n=100, quantile=0.95,
         except KeyError:
             branch_ct = {}
         branch_prev_scaling = pd.Series(
-            {i: ref_ct[i] if i in branch_ct else 1 for i in congested_indices})
+            {i: (branch_ct[i] if i in branch_ct else 1)
+            for i in congested_indices})
         branch_ratings = branch_ratings / branch_prev_scaling
     if method == 'MW':
         branch_metric = quantile_cong_abs / branch_ratings