From bb9921c402f658315c44a20a42f8eaa78693d87b Mon Sep 17 00:00:00 2001
From: Hsien-Chin Lin <linh@hhu.de>
Date: Thu, 13 Apr 2023 14:42:57 +0200
Subject: [PATCH] update

---
 convlab/policy/emoUS/dialogue_collector.py | 2 +-
 convlab/policy/emoUS/evaluate.py           | 2 --
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/convlab/policy/emoUS/dialogue_collector.py b/convlab/policy/emoUS/dialogue_collector.py
index 3422bed1..1976c0cd 100644
--- a/convlab/policy/emoUS/dialogue_collector.py
+++ b/convlab/policy/emoUS/dialogue_collector.py
@@ -95,7 +95,7 @@ if __name__ == "__main__":
                             num_goals=args.num)
     data = {"config": json.load(open(args.config)),
             "conversation": conversation}
-    folder_name = os.path.join("convlab/policy/emoTUS", "conversation")
+    folder_name = os.path.join("convlab/policy/emoUS", "conversation")
     if not os.path.exists(folder_name):
         os.makedirs(folder_name)
     json.dump(data,
diff --git a/convlab/policy/emoUS/evaluate.py b/convlab/policy/emoUS/evaluate.py
index ec1b8e44..cff2780b 100644
--- a/convlab/policy/emoUS/evaluate.py
+++ b/convlab/policy/emoUS/evaluate.py
@@ -7,8 +7,6 @@ from datetime import datetime
 import matplotlib.pyplot as plt
 import torch
 from datasets import load_metric
-# from convlab.policy.genTUS.pg.stepGenTUSagent import \
-#     stepGenTUSPG as UserPolicy
 from sklearn import metrics
 from tqdm import tqdm
 
-- 
GitLab