diff --git a/convlab/policy/emoTUS/evaluate.py b/convlab/policy/emoTUS/evaluate.py index 86af813d755ad549d44b6106cfe50018a27127c3..2204c26dcfbdcf391e4c403719c75807a76afd29 100644 --- a/convlab/policy/emoTUS/evaluate.py +++ b/convlab/policy/emoTUS/evaluate.py @@ -85,7 +85,7 @@ class Evaluator: emotion_mode = "normal" in_file = json.load(open(f_eval)) - for dialog in tqdm(in_file['dialog'][:2]): + for dialog in tqdm(in_file['dialog']): inputs = dialog["in"] labels = self.usr._parse_output(dialog["out"])