fix: fix route table total time

gnes-ai · Sep 23, 2019 · 6a6797e · 6a6797e
1 parent b0e3215
commit 6a6797e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/gnes/client/cli.py b/gnes/client/cli.py
@@ -89,13 +89,12 @@ def update(self):
         elapsed_str = colored('elapsed', 'yellow')
         speed_str = colored('speed', 'yellow')
         self.num_bars += 1
-        if self.num_bars > self.bar_len:
-            self.num_bars -= self.bar_len
+        if self.num_bars % self.bar_len == 0:
             sys.stdout.write('\n')
         sys.stdout.write(
             '{:>10} [{:<{}}]  {:>8}: {:3.1f}s   {:>8}: {:3.1f} batch/s'.format(
                 colored(self.task_name, 'cyan'),
-                colored('=' * self.num_bars, 'green'),
+                colored('=' * int(self.num_bars % self.bar_len), 'green'),
                 self.bar_len + 9,
                 elapsed_str,
                 elapsed,