diff --git a/convlab/policy/emoUS/evaluate.py b/convlab/policy/emoUS/evaluate.py index 462c1dc8d05dc21af95d5af1459d7a0bb71635f5..e6ac3968902371d99d511735b282fb76f2311436 100644 --- a/convlab/policy/emoUS/evaluate.py +++ b/convlab/policy/emoUS/evaluate.py @@ -98,7 +98,7 @@ class Evaluator: if self.sample: mode = "sample" - for dialog in tqdm(in_file['dialog'][:1]): + for dialog in tqdm(in_file['dialog']): inputs = dialog["in"] labels = self.usr._parse_output(dialog["out"])