Add suggestions from wireviz#186

kvid · Sep 12, 2023 · 162fe44 · 162fe44
1 parent c3b07cf
commit 162fe44
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/wireviz/wv_utils.py b/src/wireviz/wv_utils.py
@@ -70,9 +70,10 @@ def expand(yaml_data):
 
 
 def get_single_key_and_value(d: dict):
-    k = list(d.keys())[0]
-    v = d[k]
-    return (k, v)
+    # used for defining a line in a harness' connection set
+    # E.g. for the YAML input `- X1: 1`
+    # this function returns a tuple in the form ("X1", "1")
+    return next(iter(d.items()))
 
 
 def int2tuple(inp):