def test_cli_run_self_supervised_swav(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.self_supervised.swav.swav_module import cli_main cli_args = cli_args.split(' ') if cli_args else [] with mock.patch("argparse._sys.argv", ["any.py"] + cli_args): cli_main()
def test_cli_run_ssl_cpc(cli_args, dataset_name): from pl_bolts.models.self_supervised.cpc.cpc_module import cli_main cli_args = cli_args % {"dataset_name": dataset_name} cli_args = cli_args.strip().split(" ") if cli_args else [] with mock.patch("argparse._sys.argv", ["any.py"] + cli_args): cli_main()