diff --git a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py
index a51b3faa6..427d5ebf7 100644
--- a/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py
+++ b/tensorflow_model_optimization/python/core/sparsity/keras/pruning_callbacks.py
@@ -107,8 +107,7 @@ def _log_pruning_metrics(self, logs, prefix, step):
         self._file_writer.flush()
 
   def on_epoch_begin(self, epoch, logs=None):
-    if logs is not None:
-      super(PruningSummaries, self).on_epoch_begin(epoch, logs)
+    del logs
 
     pruning_logs = {}
     params = []