clean up debug prints

Author: DNXie

src/forge/observability/metric_actors.py

@@ -433,7 +433,6 @@ def extract_values_from_valuemesh(results) -> list[dict[str, Any]]:
             # Reduce metrics from states
             reduced_metrics = reduce_metrics_states(all_local_states)
 
-            print(f"[DEBUG] reduced_metrics: {reduced_metrics}")
             # Split into scalar metrics and sample metrics
             scalar_metrics = [
                 m for m in reduced_metrics if m.reduction != Reduce.SAMPLE
@@ -445,7 +444,6 @@ def extract_values_from_valuemesh(results) -> list[dict[str, Any]]:
             # Log to global backends
             for backend_name, backend in self.global_logger_backends.items():
                 if scalar_metrics:
-                    print(f"[DEBUG] calling log_batch from GlobalLoggerActor")
                     await backend.log_batch(scalar_metrics, global_step)
                 if sample_metrics:
                     await backend.log_samples(sample_metrics, global_step)

src/forge/observability/metrics.py

@@ -228,18 +228,7 @@ def record_episode_sample(table_name: str, episode):
         "response_len": episode.response_len,
         "pad_id": episode.pad_id,
     }
-
-    print(
-        "[DEBUG] Adding sample to table via record_metric, episode_id: ",
-        episode.episode_id,
-        # "episode: ",
-        # episode,
-    )
     record_metric(table_name, sample, Reduce.SAMPLE)
-    print(
-        "[DEBUG] Added sample to table via record_metric, episode_id: ",
-        episode.episode_id,
-    )
 
 
 #################
@@ -789,7 +778,6 @@ async def flush(
 
             for backend in self.per_rank_reduce_backends:
                 if scalar_metrics:
-                    print(f"[DEBUG] calling log_batch from MetricCollector")
                     await backend.log_batch(scalar_metrics, global_step)
                 if sample_metrics:
                     await backend.log_samples(sample_metrics, global_step)
@@ -912,7 +900,6 @@ async def init(
     async def log_batch(
         self, metrics: list[Metric], global_step: int, *args, **kwargs
     ) -> None:
-        print(f"[DEBUG] calling log_batch with {len(metrics)} metrics")
         metrics_str = "\n".join(
             f"  {metric.key}: {metric.value}"
             for metric in sorted(metrics, key=lambda m: m.key)
@@ -931,12 +918,10 @@ async def log_samples(self, samples: Dict[str, List[dict]], step: int) -> None:
         """Pretty-print sample-level logs to console."""
         import json
 
-        print(f"[DEBUG] calling log_samples with {len(samples)} samples")
-
         logger.info(f"==========  SAMPLE LOGS STEP {step} ==========")
         for table_name, table_rows in samples.items():
             logger.info(f"[{table_name}] ({len(table_rows)} samples)")
-            logger.info(json.dumps(table_rows, indent=2, ensure_ascii=False))
+            logger.info(json.dumps(table_rows))
         logger.info("==============================================\n")