Merge pull request #697 from NVIDIA/am/ench

Small formatting improvements

Author: amaslenn

src/cloudai/cli/handlers.py

@@ -150,10 +150,11 @@ def handle_dse_job(runner: Runner, args: argparse.Namespace) -> int:
                 break
             step, action = result
             env.test_run.step = step
+            logging.info(f"Running step {step} (of {agent.max_steps}) with action {action}")
             observation, reward, done, info = env.step(action)
             feedback = {"trial_index": step, "value": reward}
             agent.update_policy(feedback)
-            logging.info(f"Step {step}: Observation: {observation}, Reward: {reward}")
+            logging.info(f"Step {step}: Observation: {[round(obs, 4) for obs in observation]}, Reward: {reward:.4f}")
 
     if args.mode == "run":
         runner.runner.test_scenario.test_runs = original_test_runs

src/cloudai/configurator/cloudai_gym.py

@@ -104,7 +104,6 @@ def step(self, action: Any) -> Tuple[list, float, bool, dict]:
             logging.info("Constraint check failed. Skipping step.")
             return [-1.0], -1.0, True, {}
 
-        logging.info(f"Running step {self.test_run.step} with action {action}")
         new_tr = copy.deepcopy(self.test_run)
         new_tr.output_path = self.runner.get_job_output_path(new_tr)
         self.runner.test_scenario.test_runs = [new_tr]

src/cloudai/workloads/nixl_bench/nixl_summary_report.py

@@ -78,7 +78,7 @@ def create_charts(self, cmp_groups: list[GroupedTestRuns]) -> list[bk.figure]:
             dfs = [self.extract_data_as_df(item.tr) for item in group.items]
             charts.extend(
                 [
-                    self.create_chart(group, dfs, "Latecy", list(self.INFO_COLUMNS), ["avg_lat"], "Time (us)"),
+                    self.create_chart(group, dfs, "Latency", list(self.INFO_COLUMNS), ["avg_lat"], "Time (us)"),
                     self.create_chart(group, dfs, "Bandwidth", list(self.INFO_COLUMNS), ["bw_gb_sec"], "Busbw (GB/s)"),
                 ]
             )