def test_cli_run_log_regression(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.regression.logistic_regression 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_retinanet(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.detection.retinanet.retinanet_module 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_basic_vae(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.autoencoders.basic_vae.basic_vae_module 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_vision_image_gpt(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.vision.image_gpt.igpt_module 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_basic_gan(cli_args, dataset_name): from pl_bolts.models.gans.basic.basic_gan_module import cli_main cli_args = cli_args % {'dataset_name': dataset_name} with mock.patch("argparse._sys.argv", ["any.py"] + cli_args.strip().split()): cli_main()
def test_cli_run_mnist(cli_args): """Test running CLI for an example with default params.""" from pl_bolts.models.mnist_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_basic_ae(dataset_name): from pl_bolts.models.autoencoders.basic_ae.basic_ae_module import cli_main cli_args = f""" --dataset {dataset_name} --max_epochs 1 --limit_train_batches 3 --limit_val_batches 3 --batch_size 3 """.strip().split() with mock.patch("argparse._sys.argv", ["any.py"] + cli_args): cli_main()
def test_cli_cpc(cli_args): from pl_bolts.models.self_supervised.cpc.cpc_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_basic_gan(cli_args): from pl_bolts.models.gans.basic.basic_gan_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_srresnet(cli_args): from pl_bolts.models.gans.srgan.srresnet_module import cli_main with mock.patch("argparse._sys.argv", ["any.py"] + cli_args.strip().split()): cli_main()
def test_cli_run_basic_gan(cli_args): from pl_bolts.models.gans.basic.basic_gan_module import cli_main with mock.patch("argparse._sys.argv", ["any.py"] + cli_args.strip().split()): cli_main()