Skip to content

Commit 280bda1

Browse files
committed
fix comment
1 parent 19dc713 commit 280bda1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

trinity/utils/monitor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def gather_eval_metrics(
6969
for col in numeric_df.columns:
7070
# Skip the columns that are already aggregated
7171
key_words = ["std", "mean", "min", "max"]
72-
if any(key_word in col.lower() for key_word in key_words):
72+
if any(col.endswith(key_word) for key_word in key_words):
7373
metric[f"{prefix}/{col}"] = numeric_df[col].mean()
7474
else:
7575
stats_df = numeric_df[[col]].agg(output_stats)

0 commit comments

Comments
 (0)