Exemplos de cli_main em Python

Exemplo n.º 1

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()

Exemplo n.º 2

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()