Check reporthook is not None before calling it (keras-team#8320)

mgq1507 · Oct 31, 2017 · abccbba · abccbba
1 parent ee4b0bc
commit abccbba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/keras/utils/data_utils.py b/keras/utils/data_utils.py
@@ -57,7 +57,8 @@ def chunk_read(response, chunk_size=8192, reporthook=None):
                 chunk = response.read(chunk_size)
                 count += 1
                 if not chunk:
-                    reporthook(count, total_size, total_size)
+                    if reporthook:
+                        reporthook(count, total_size, total_size)
                     break
                 if reporthook:
                     reporthook(count, chunk_size, total_size)