Merge pull request #726 from NVIDIA/am/path-in-stdout

Report results dir to users as early as possible

Author: amaslenn

src/cloudai/_core/base_runner.py

@@ -104,8 +104,8 @@ async def submit_test(self, tr: TestRun):
         Args:
             tr (TestRun): The test to be started.
         """
-        logging.info(f"Starting test: {tr.name}")
         tr.output_path = self.get_job_output_path(tr)
+        logging.info(f"Starting test: {tr.name} (results at: {tr.output_path})")
         self.on_job_submit(tr)
         try:
             job = self._submit_test(tr)
@@ -126,7 +126,7 @@ async def delayed_submit_test(self, tr: TestRun, delay: int = 5):
             tr (TestRun): The test to start after a delay.
             delay (int): Delay in seconds before starting the test.
         """
-        logging.info(f"Delayed start for test {tr.name} by {delay} seconds.")
+        logging.debug(f"Delayed start for test {tr.name} by {delay} seconds.")
         await asyncio.sleep(delay)
         await self.submit_test(tr)
 

src/cloudai/cli/handlers.py

@@ -189,8 +189,6 @@ def generate_reports(system: System, test_scenario: TestScenario, result_dir: Pa
 def handle_non_dse_job(runner: Runner, args: argparse.Namespace) -> None:
     asyncio.run(runner.run())
 
-    logging.info(f"All test scenario results stored at: {runner.runner.scenario_root}")
-
     if args.mode == "run":
         generate_reports(runner.runner.system, runner.runner.test_scenario, runner.runner.scenario_root)
 
@@ -294,6 +292,7 @@ def handle_dry_run_and_run(args: argparse.Namespace) -> int:
 
     runner = Runner(args.mode, system, test_scenario)
     register_signal_handlers(runner.cancel_on_signal)
+    logging.info(f"Scenario results will be stored at: {runner.runner.scenario_root}")
 
     has_dse = any(tr.is_dse_job for tr in test_scenario.test_runs)
     if args.single_sbatch or not has_dse:  # in this mode cases are unrolled using grid search