Skip to content

Commit 404ee1d

Browse files
author
Felipe Mello
committed
improve log msg
1 parent f39a9f9 commit 404ee1d

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/forge/observability/metrics.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -507,7 +507,8 @@ def push(self, metric: Metric) -> None:
507507
logger,
508508
level=logging.WARNING,
509509
msg=(
510-
"Skipping metric collection. Metric logging backends (e.g. wandb) were not initialized."
510+
f"Skipping metric collection for {get_proc_name_with_rank()}."
511+
" Metric logging backends (e.g. wandb) were not initialized."
511512
" This happens when you try to use `record_metric` before calling `init_backends`."
512513
" To disable this warning, please call in your main file:\n"
513514
"`mlogger = await get_or_create_metric_logger()`\n"
@@ -546,7 +547,8 @@ async def flush(
546547
log_once(
547548
logger,
548549
level=logging.WARNING,
549-
msg="Cannot flush collected metrics. MetricCollector.flush() called before init_backends()."
550+
msg=f"Cannot flush collected metrics for {get_proc_name_with_rank()}. "
551+
" MetricCollector.flush() called before init_backends()."
550552
"\nPlease call in your main file:\n"
551553
"`mlogger = await get_or_create_metric_logger()`\n"
552554
"`await mlogger.init_backends.call_one(logging_config)`\n"

0 commit comments

Comments
 (0)