From 80f24c0c2ed74312494461574b7667911fdb9a40 Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Fri, 14 Apr 2023 15:33:41 +0200
Subject: [PATCH] wip
---
convlab/policy/emoUS/evaluate.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/convlab/policy/emoUS/evaluate.py b/convlab/policy/emoUS/evaluate.py
index 462c1dc8..e6ac3968 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"])
--
GitLab