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