resolve pandas future warning, integer keys

kaklise · May 7, 2024 · 1df8f3c · 1df8f3c
1 parent bda99a2
commit 1df8f3c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/wntr/metrics/topographic.py b/wntr/metrics/topographic.py
@@ -232,9 +232,7 @@ def valve_segments(G, valve_layer):
             seg_label[all_names.index(node_name)] = seg_index
 
     # Collect valved link names
-    valved_link_names = []    
-    for i, row in valve_layer.iterrows():
-        valved_link_names.append(row[0])
+    valved_link_names = list(valve_layer['link'].unique()) 
 
     # Remove valved edges from G
     valved_edges = []