diff --git a/visualdl/server/lib.py b/visualdl/server/lib.py index 00ee95506686f50496994cae8c040b1d45b1f1a2..8f273d6c5baa3f30ec2d620f0945b4e67df90153 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))