Unverified Commit f099249c authored by Boris Dayma's avatar Boris Dayma Committed by GitHub
Browse files

fix(integrations): consider test metrics (#13888)

parent 0ddadbf0
...@@ -358,9 +358,13 @@ def rewrite_logs(d): ...@@ -358,9 +358,13 @@ def rewrite_logs(d):
new_d = {} new_d = {}
eval_prefix = "eval_" eval_prefix = "eval_"
eval_prefix_len = len(eval_prefix) eval_prefix_len = len(eval_prefix)
test_prefix = "test_"
test_prefix_len = len(test_prefix)
for k, v in d.items(): for k, v in d.items():
if k.startswith(eval_prefix): if k.startswith(eval_prefix):
new_d["eval/" + k[eval_prefix_len:]] = v new_d["eval/" + k[eval_prefix_len:]] = v
elif k.startswith(test_prefix):
new_d["test/" + k[test_prefix_len:]] = v
else: else:
new_d["train/" + k] = v new_d["train/" + k] = v
return new_d return new_d
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment