File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments