summaryrefslogtreecommitdiffstats
path: root/tqdm/keras.py
diff options
context:
space:
mode:
Diffstat (limited to 'tqdm/keras.py')
-rw-r--r--tqdm/keras.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/tqdm/keras.py b/tqdm/keras.py
index 523e62e..cce9467 100644
--- a/tqdm/keras.py
+++ b/tqdm/keras.py
@@ -1,5 +1,3 @@
-from __future__ import absolute_import, division
-
from copy import copy
from functools import partial
@@ -96,7 +94,7 @@ class TqdmCallback(keras.callbacks.Callback):
raise KeyError('Unknown verbosity')
def on_train_end(self, *_, **__):
- if self.verbose:
+ if hasattr(self, 'batch_bar'):
self.batch_bar.close()
self.epoch_bar.close()