From 81d5b6d06a47168874629c2444c73e1d4c34f070 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 09:49:43 +0100
Subject: use util func

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

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

diff --git a/20/py/aoc20.py b/20/py/aoc20.py
index c5e50c1..15a8782 100644
--- a/20/py/aoc20.py
+++ b/20/py/aoc20.py
@@ -111,10 +111,10 @@ if __name__ == "__main__":
         print("                ---------", end="")
     print()
     print("avg   ", end="")
-    if only_part in (0, 1):
+    if running_part(1):
         print(f"{tot_time_part[0]*1000/runs:6.3f}", end="")
     if only_part == 0:
         print("                   ", end="")
-    if only_part in (0, 2):
+    if running_part(2):
         print(f"{tot_time_part[1]*1000/runs:6.3f}", end="")
     print()
-- 
cgit v1.2.1