Skip to content

Commit d2800cf

Browse files
committed
v2: Update test names for Reinforce
1 parent 9ba76b7 commit d2800cf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/discrete/test_reinforce.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def get_args() -> argparse.Namespace:
4747
return parser.parse_known_args()[0]
4848

4949

50-
def test_pg(args: argparse.Namespace = get_args(), enable_assertions: bool = True) -> None:
50+
def test_reinforce(args: argparse.Namespace = get_args(), enable_assertions: bool = True) -> None:
5151
env = gym.make(args.task)
5252
space_info = SpaceInfo.from_env(env)
5353
args.state_shape = space_info.observation_info.obs_shape
@@ -135,6 +135,6 @@ def stop_fn(mean_rewards: float) -> bool:
135135
assert stop_fn(result.best_reward)
136136

137137

138-
def test_pg_determinism() -> None:
139-
main_fn = lambda args: test_pg(args, enable_assertions=False)
138+
def test_reinforce_determinism() -> None:
139+
main_fn = lambda args: test_reinforce(args, enable_assertions=False)
140140
AlgorithmDeterminismTest("discrete_reinforce", main_fn, get_args()).run()

0 commit comments

Comments
 (0)