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

This commit is contained in:
Boris Dayma 2021-10-05 15:27:22 -05:00 committed by GitHub
parent 0ddadbf0a8
commit f099249cf1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -358,9 +358,13 @@ def rewrite_logs(d):
new_d = {}
eval_prefix = "eval_"
eval_prefix_len = len(eval_prefix)
test_prefix = "test_"
test_prefix_len = len(test_prefix)
for k, v in d.items():
if k.startswith(eval_prefix):
new_d["eval/" + k[eval_prefix_len:]] = v
elif k.startswith(test_prefix):
new_d["test/" + k[test_prefix_len:]] = v
else:
new_d["train/" + k] = v
return new_d