From 7b63aa4f1a4218ae6b48261b96e9c38a9ce35336 Mon Sep 17 00:00:00 2001
From: ShenYuhan <yhan_shen@163.com>
Date: Fri, 18 Sep 2020 18:06:24 +0800
Subject: [PATCH] fix pr display_name bug
---
visualdl/server/lib.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/visualdl/server/lib.py b/visualdl/server/lib.py
index 00ee9550..8f273d6c 100644
--- a/visualdl/server/lib.py
+++ b/visualdl/server/lib.py
@@ -169,9 +169,10 @@ def get_pr_curve(log_reader, run, tag):
def get_pr_curve_step(log_reader, run, tag=None):
+ fake_run = run
run = log_reader.name2tags[run] if run in log_reader.name2tags else run
run2tag = get_pr_curve_tags(log_reader)
- tag = run2tag['tags'][run2tag['runs'].index(run)][0]
+ tag = run2tag['tags'][run2tag['runs'].index(fake_run)][0]
log_reader.load_new_data()
records = log_reader.data_manager.get_reservoir("pr_curve").get_items(
run, decode_tag(tag))
--
GitLab