Skip to content

Commit c6c74dc

Browse files
committed
reset from argument
1 parent e8679d2 commit c6c74dc

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

test/pettingzoo/tic_tac_toe.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -235,6 +235,5 @@ def watch(
235235
policy, optim, agents = get_agents(args, agent_learn=agent_learn, agent_opponent=agent_opponent)
236236
policy.policies[agents[args.agent_id - 1]].set_eps(args.eps_test)
237237
collector = Collector[CollectStats](policy, env, exploration_noise=True)
238-
collector.reset()
239-
result = collector.collect(n_episode=1, render=args.render)
238+
result = collector.collect(n_episode=1, render=args.render, reset_before_collect=True)
240239
result.pprint_asdict()

0 commit comments

Comments
 (0)