Fix format seconds for days

Change-Id: Ib7001ff2ed96f16b03fc6a043aca6c7708b9308b
Jeffwhen · Mar 13, 2023 · 39e3cd7 · 39e3cd7
1 parent 947254a
commit 39e3cd7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/tpu_perf/util.py b/python/tpu_perf/util.py
@@ -29,8 +29,10 @@ def dict_override(a, b):
 
 from datetime import timedelta
 def format_seconds(s):
-    delta = timedelta(seconds=s)
+    days = s // (24 * 60 * 60)
+    delta = timedelta(seconds=s % (24 * 60 * 60))
     pairs = zip(
         [int(v.split('.')[0]) for v in str(delta).split(':')],
         ['hours', 'minutes', 'seconds'])
-    return ' '.join(f'{v} {u}' for v, u in pairs if v) or '0 second'
+    ret = ' '.join(f'{v} {u}' for v, u in pairs if v) or '0 second'
+    return f'{days} days {ret}' if days else ret