()
| 88 | print(f"Skip over this item due to error {str(e)}") |
| 89 | |
| 90 | def main(): |
| 91 | parser = argparse.ArgumentParser(description="Generate OptILM dataset") |
| 92 | parser.add_argument("--num_samples", type=int, default=100, help="Number of samples to process") |
| 93 | parser.add_argument("--output_file", type=str, default="optillm_dataset.jsonl", help="Output file path") |
| 94 | args = parser.parse_args() |
| 95 | |
| 96 | asyncio.run(generate_dataset(args.num_samples, args.output_file)) |
| 97 | print(f"Dataset generated and saved to {args.output_file}") |
| 98 | |
| 99 | if __name__ == "__main__": |
| 100 | main() |
no test coverage detected