def test_cli_run_rl_per_dqn(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.rl.per_dqn_model import cli_main cli_args = cli_args.strip().split(" ") if cli_args else [] with mock.patch("argparse._sys.argv", ["any.py"] + cli_args): cli_main()
def test_cli_run_rl_vanilla_policy_gradient(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.rl.vanilla_policy_gradient_model import cli_main cli_args = cli_args.strip().split(' ') if cli_args else [] with mock.patch("argparse._sys.argv", ["any.py"] + cli_args): cli_main()