ci: Fix stopiteration for Mbridge (#2760)

ko3n1g · web-flow · commit ca66cdc2615c · 2026-03-11T20:54:39.000+01:00
Signed-off-by: oliver könig &lt;okoenig@nvidia.com&gt;
diff --git a/scripts/performance/setup_experiment.py b/scripts/performance/setup_experiment.py
@@ -73,6 +73,7 @@ def check_training_finished(log_file_paths: List[str]) -> bool:
                         "StopIteration" in line
                         or "after training is done" in line
                         or "exiting program at iteration" in line
+                        or "AssertionError: no samples left to consume:" in line
                     )
                 )
     return any(all_lines)

Original file line number	Diff line number	Diff line change
`@@ -73,6 +73,7 @@ def check_training_finished(log_file_paths: List[str]) -> bool:`
`73`	`73`	`"StopIteration" in line`
`74`	`74`	`or "after training is done" in line`
`75`	`75`	`or "exiting program at iteration" in line`
	`76`	`+ or "AssertionError: no samples left to consume:" in line`
`76`	`77`	`)`
`77`	`78`	`)`
`78`	`79`	`return any(all_lines)`