truncate the visualization to < 100

Signed-off-by: Youngeun Kwon <[email protected]>

Author: youngeunkwon0405

nemo_rl/algorithms/utils.py

@@ -457,6 +457,7 @@ def visualize_per_worker_timeline(
         timeline_interval: float | None,
     ) -> None:
         dp_ranks = list(metric_dict.keys())
+        max_rows_to_print = 100
         max_timeline_length = 50
         marker = {0: "▃", 1: "▅", 2: "▆", 3: "▉"}
         zero_marker = "▁"
@@ -471,6 +472,8 @@ def visualize_per_worker_timeline(
             f"    - Timeline (0: {zero_marker}, {', '.join(f'{1.0 if k == 0 else k * (max_value / len(marker))}-{(k + 1) * (max_value / len(marker))}: {marker[k]}' for k in marker.keys())}):"
         )
         for dp_idx, metric_values in metric_dict.items():
+            if dp_idx > max_rows_to_print:
+                break
             timeline = []
             length = len(metric_values)
             if timeline_interval is not None: