Add suggestions from wireviz#186

kvid · Aug 27, 2023 · 8c7ef40 · 8c7ef40
1 parent 1e78b3c
commit 8c7ef40
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):