Skip to content

Commit 4ac407c

Browse files
Remove test_fn and train_fn as they are not used in PPO PistonBall example for PettingZoo (#840)
Specifically, BasePolicy.set_eps seems to be a remnant from using DQN in other examples. * Removed unused functions (test_fn and train_fn) from the pettingzoo example with PistonBall. These functions use set_eps which is not available for PPO and is not even called once in the file.
1 parent 7f8fa24 commit 4ac407c

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

test/pettingzoo/pistonball_continuous.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -237,12 +237,6 @@ def save_best_fn(policy):
237237
def stop_fn(mean_rewards):
238238
return False
239239

240-
def train_fn(epoch, env_step):
241-
[agent.set_eps(args.eps_train) for agent in policy.policies.values()]
242-
243-
def test_fn(epoch, env_step):
244-
[agent.set_eps(args.eps_test) for agent in policy.policies.values()]
245-
246240
def reward_metric(rews):
247241
return rews[:, 0]
248242

0 commit comments

Comments
 (0)