From 15a72a4f3a6e7b9276f3f501f3d29ee2d92666eb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gustav=20S=C3=B6rn=C3=A4s?= <gustav@sornas.net>
Date: Wed, 9 Dec 2020 07:58:32 +0100
Subject: add driver

---
 20/py/d07.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

(limited to '20/py/d07.py')

diff --git a/20/py/d07.py b/20/py/d07.py
index 48fa3d2..f96c796 100644
--- a/20/py/d07.py
+++ b/20/py/d07.py
@@ -11,7 +11,7 @@ class Node:
         self.children = children
 
     def __iter__(self):
-        yield from child for child, _ in self.children
+        yield from (child for child, _ in self.children)
 
 
 def parse(_in):
@@ -50,7 +50,5 @@ if __name__ == "__main__":
         input = open("../input/07", "r").readlines()
     # graph(parse(input))
 
-    start = time.time()
-    print(pt2(input))
     print(pt1(input))
-    print(time.time() - start)
+    print(pt2(input))
-- 
cgit v1.2.1