From d12974882fb5cedf48df05ad82bb0874bb3ffcf9 Mon Sep 17 00:00:00 2001
From: ShenYuhan <yhan_shen@163.com>
Date: Wed, 19 Aug 2020 09:40:51 +0800
Subject: [PATCH] fix bug (#760)

---
 visualdl/server/lib.py | 3 ++-
 visualdl/version.py    | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/visualdl/server/lib.py b/visualdl/server/lib.py
index 7976fd45b..660b1ae6b 100644
--- a/visualdl/server/lib.py
+++ b/visualdl/server/lib.py
@@ -166,7 +166,8 @@ def get_pr_curve(log_reader, run, tag):
 
 def get_pr_curve_step(log_reader, run, tag=None):
     run = log_reader.name2tags[run] if run in log_reader.name2tags else run
-    tag = get_pr_curve_tags(log_reader)[run][0] if tag is None else tag
+    run2tag = get_pr_curve_tags(log_reader)
+    tag = run2tag['tags'][run2tag['runs'].index(run)][0]
     log_reader.load_new_data()
     records = log_reader.data_manager.get_reservoir("pr_curve").get_items(
         run, decode_tag(tag))
diff --git a/visualdl/version.py b/visualdl/version.py
index 0bcb49fe1..16511fb0f 100644
--- a/visualdl/version.py
+++ b/visualdl/version.py
@@ -13,4 +13,4 @@
 # limitations under the License.
 # =======================================================================
 
-vdl_version = '2.0.0'
+vdl_version = '2.0.1'