Skip to content

Commit 7fc5466

Browse files
committed
Fix progbar dynamic display
1 parent 3fe5254 commit 7fc5466

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

keras/utils/generic_utils.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -266,8 +266,9 @@ def __init__(self, target, width=30, verbose=1, interval=0.05):
266266
self.interval = interval
267267
self.total_width = 0
268268
self.seen_so_far = 0
269-
self.verbose = verbose
270-
self.is_jupyter = 'ipykernel' in sys.modules
269+
self.verbose = verbose
270+
self._dynamic_display = (sys.stdout.isatty() or
271+
'ipykernel' in sys.modules)
271272

272273
def update(self, current, values=None, force=False):
273274
"""Updates the progress bar.
@@ -297,7 +298,7 @@ def update(self, current, values=None, force=False):
297298
return
298299

299300
prev_total_width = self.total_width
300-
if sys.stdout.isatty() or self.is_jupyter:
301+
if self._dynamic_display:
301302
sys.stdout.write('\b' * prev_total_width)
302303
sys.stdout.write('\r')
303304
else:
@@ -348,7 +349,8 @@ def update(self, current, values=None, force=False):
348349
for k in self.unique_values:
349350
info += ' - %s:' % k
350351
if isinstance(self.sum_values[k], list):
351-
avg = np.mean(self.sum_values[k][0] / max(1, self.sum_values[k][1]))
352+
avg = np.mean(
353+
self.sum_values[k][0] / max(1, self.sum_values[k][1]))
352354
if abs(avg) > 1e-3:
353355
info += ' %.4f' % avg
354356
else:
@@ -370,7 +372,8 @@ def update(self, current, values=None, force=False):
370372
if self.target is None or current >= self.target:
371373
for k in self.unique_values:
372374
info += ' - %s:' % k
373-
avg = np.mean(self.sum_values[k][0] / max(1, self.sum_values[k][1]))
375+
avg = np.mean(
376+
self.sum_values[k][0] / max(1, self.sum_values[k][1]))
374377
if avg > 1e-3:
375378
info += ' %.4f' % avg
376379
else:

0 commit comments

Comments
 (0)