From 32ff8b4a6a5a836065809f293ecf06d4118624d5 Mon Sep 17 00:00:00 2001
From: Daniel Olsen <dolsen@intven.com>
Date: Sat, 4 Apr 2020 19:12:35 -0700
Subject: [PATCH] fix: correct bug when MW-mesh scaling with existing ct

---
 powersimdata/input/design_transmission.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/powersimdata/input/design_transmission.py b/powersimdata/input/design_transmission.py
index ac76b3d84..69dfb88fa 100644
--- 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