diff --git a/convlab2/policy/evaluate.py b/convlab2/policy/evaluate.py index 8085bf40d140bca2dcaf6af9a15b49ebdcee21ff..de0a06fa24aa598205fc4b98d3118f81feb9db09 100755 --- a/convlab2/policy/evaluate.py +++ b/convlab2/policy/evaluate.py @@ -222,8 +222,7 @@ def evaluate(dataset_name, model_name, load_path, calculate_reward=True): for key in sess.evaluator.goal: if key not in task_success: task_success[key] = [] - else: - task_success[key].append(task_succ) + task_success[key].append(task_succ) task_success['All'].append(task_succ) for key in task_success: