[ViPPET] Fix max_runtime propagation in density tests (open-edge-platform#1888)

Author: ktaube26

tools/visual-pipeline-and-platform-evaluation-tool/vippet/benchmark.py

@@ -54,10 +54,10 @@ def __repr__(self):
 class Benchmark:
     """Benchmarking class for pipeline evaluation."""
 
-    def __init__(self):
+    def __init__(self, max_runtime: float = 0):
         self.best_result = None
-        # Initialize PipelineRunner in normal mode with max_runtime=0 (run until EOS)
-        self.runner = PipelineRunner(mode="normal", max_runtime=0)
+        # Initialize PipelineRunner in normal mode with optional max_runtime for each run
+        self.runner = PipelineRunner(mode="normal", max_runtime=max_runtime)
         self.logger = logging.getLogger(__name__)
 
     @staticmethod

tools/visual-pipeline-and-platform-evaluation-tool/vippet/managers/tests_manager.py

@@ -454,7 +454,9 @@ def _execute_density_test(
             )
 
             # Initialize Benchmark
-            benchmark = Benchmark()
+            benchmark = Benchmark(
+                max_runtime=internal_spec.execution_config.max_runtime
+            )
 
             # Store benchmark runner for this job so that a future extension could cancel it.
             with self._jobs_lock: