Internal

PiperOrigin-RevId: 710775406
google · Dec 30, 2024 · 7e90b81 · 7e90b81
1 parent 6d6fa74
commit 7e90b81
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/grain/_src/python/dataset/stats.py b/grain/_src/python/dataset/stats.py
@@ -532,9 +532,9 @@ def _logging_execution_summary_loop(self):
             " the `PrefetchDatasetIterator` node indicates it is a bottleneck."
             " The `MapDataset` nodes are executed in multiple threads and thus,"
             " should not be compared to the `total_processing_time` of"
-            " `DatasetIterator` nodes.."
+            " `DatasetIterator` nodes.\n\n%s",
+            _pretty_format_summary(summary),
         )
-        logging.info(_pretty_format_summary(summary))
 
   def _build_execution_summary(
       self,