Fix the peak memory usage calculation (ray-project#28419)

justinvyu · Sep 14, 2022 · 0717baa · 0717baa
1 parent 2a4064c
commit 0717baa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/ray/data/_internal/stats.py b/python/ray/data/_internal/stats.py
@@ -335,7 +335,9 @@ def _summarize_blocks(self, blocks: List[BlockMetadata], is_substage: bool) -> s
             )
 
             out += indent
-            memory_stats = [round(e.max_rss_bytes / 1024 * 1024, 2) for e in exec_stats]
+            memory_stats = [
+                round(e.max_rss_bytes / (1024 * 1024), 2) for e in exec_stats
+            ]
             out += "* Peak heap memory usage (MiB): {} min, {} max, {} mean\n".format(
                 min(memory_stats),
                 max(memory_stats),