keras: fix tensorflow==2.2.0 compatibility

- fixes #979
tqdm · Jun 3, 2020 · 133b218 · 133b218
1 parent 554b2a3
commit 133b218
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tqdm/keras.py b/tqdm/keras.py
@@ -91,3 +91,12 @@ def on_train_end(self, *_, **__):
         if self.verbose:
             self.batch_bar.close()
         self.epoch_bar.close()
+
+    def _implements_train_batch_hooks(self):
+        return True
+
+    def _implements_test_batch_hooks(self):
+        return True
+
+    def _implements_predict_batch_hooks(self):
+        return True