Skip to content
Snippets Groups Projects
Unverified Commit 0bd551b5 authored by liangrz15's avatar liangrz15 Committed by GitHub
Browse files

fix reset to make user talk first (#169)

parent 2746ba72
Branches
Tags
No related merge requests found
Show whitespace changes
Inline Side-by-side
convlab2/dialog_agent/env.py
+ 1
− 0
View file @ 0bd551b5
...@@ -19,6 +19,7 @@ class Environment(): ...@@ -19,6 +19,7 @@ class Environment():
self.sys_dst.init_session() self.sys_dst.init_session()
if self.evaluator: if self.evaluator:
self.evaluator.add_goal(self.usr.policy.get_goal()) self.evaluator.add_goal(self.usr.policy.get_goal())
s, r, t = self.step([])
return self.sys_dst.state return self.sys_dst.state
def step(self, action): def step(self, action):
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment