From eaf57a81bc4637c0df4c0321c9ac3d1eee5e8482 Mon Sep 17 00:00:00 2001
From: Daniel Rojas <github@danielrojas.net>
Date: Wed, 27 May 2020 18:50:27 +0200
Subject: [PATCH] Re-format code

oops
---
 src/yaml2wireviz.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/yaml2wireviz.py b/src/yaml2wireviz.py
index 60dbaa9..01a452c 100644
--- a/src/yaml2wireviz.py
+++ b/src/yaml2wireviz.py
@@ -47,10 +47,19 @@ with open(filename, 'r') as stream:
 h = wireviz.Harness()
 # add nodes
 for k, o in input['nodes'].items():
-    h.add_node(k, type=o.get('type'), gender=o.get('gender'), num_pins=o.get('num_pins'), pinout=o.get('pinout'))
+    h.add_node(k, type=o.get('type'),
+                  gender=o.get('gender'),
+                  num_pins=o.get('num_pins'),
+                  pinout=o.get('pinout'))
 # add wires
 for k, o in input['wires'].items():
-    h.add_cable(k, mm2=o.get('mm2'), length=o.get('length'), num_wires=o.get('num_wires'), color_code=o.get('color_code'), shield=o.get('shield'))
+    h.add_cable(k, mm2=o.get('mm2'),
+                   awg=o.get('awg'),
+                   length=o.get('length'),
+                   num_wires=o.get('num_wires'),
+                   colors=o.get('colors'),
+                   color_code=o.get('color_code'),
+                   shield=o.get('shield'))
 # add connections
 conlist = input['connections']
 for con in conlist: