def test_cli(): cli_args = ["flash", "graph_classification", "--trainer.fast_dev_run", "True"] with mock.patch("sys.argv", cli_args): try: main() except SystemExit: pass
def test_cli(): cli_args = ["flash", "instance_segmentation", "--trainer.fast_dev_run", "True"] with mock.patch("sys.argv", cli_args): try: main() except SystemExit: pass
def test_cli(): cli_args = ["flash", "style_transfer", "--trainer.fast_dev_run", "True"] with mock.patch("sys.argv", cli_args): try: main() except SystemExit: pass
def _test_cli(self, extra_args: List): """Tests that the default Flash zero configuration runs for the task.""" cli_args = ["flash", self.cli_command, "--trainer.fast_dev_run", "True"] + extra_args with mock.patch("sys.argv", cli_args): try: main() except SystemExit: pass
def test_cli(): cli_args = [ "flash", "question_answering", "--trainer.fast_dev_run", "True" ] with mock.patch("sys.argv", cli_args): try: main() except SystemExit: pass
def test_cli(): cli_args = [ "flash", "speech_recognition", "--trainer.fast_dev_run", "True" ] with mock.patch("sys.argv", cli_args): try: main() except SystemExit: pass
def test_cli(cli_args): with mock.patch("sys.argv", cli_args): try: main() except SystemExit: pass