def run_benchmark_gpt2(self, arguments: str): from benchmark_gpt2 import parse_arguments, main args = parse_arguments(arguments.split()) csv_filename = main(args) self.assertTrue(os.path.exists(csv_filename))
def run_benchmark_gpt2(self, arguments: str): args = parse_arguments(arguments.split()) csv_filename = main(args) self.assertIsNotNone(csv_filename) self.assertTrue(os.path.exists(csv_filename))