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.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.split(' ') if cli_args else [] with mock.patch("argparse._sys.argv", ["any.py"] + cli_args): cli_main()
def test_cli_run_gan(cli_args): """Test running CLI for an example with default params.""" 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()