refactor: connect transformers in 'cascade' rather than 'tree'

Breakthrough-Energy · Feb 25, 2022 · b7d62bb · b7d62bb
1 parent 5159e25
commit b7d62bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/prereise/gather/griddata/hifld/data_process/transmission.py b/prereise/gather/griddata/hifld/data_process/transmission.py
@@ -517,9 +517,9 @@ def create_transformers(bus):
         ["from_bus_id", "to_bus_id"].
     """
     bus_pairs = [
-        (b, volt_series.idxmax())
+        (b, volt_series.sort_values().index[i + 1])
         for sub, volt_series in bus.groupby("sub_id")["baseKV"]
-        for b in volt_series.sort_values().index[:-1]
+        for i, b in enumerate(volt_series.sort_values().index[:-1])
         if len(volt_series) > 1
     ]