(argv)
| 425 | |
| 426 | |
| 427 | def main(argv): |
| 428 | JaxDistributedConfig.initialize(FLAGS.jax_distributed) |
| 429 | set_random_seed(FLAGS.seed) |
| 430 | |
| 431 | ht = LLMNeedleHaystackTester( |
| 432 | haystack_file=FLAGS.haystack_file, |
| 433 | context_lengths_min=FLAGS.context_lengths_min, |
| 434 | context_lengths_max=FLAGS.context_lengths_max, |
| 435 | context_lengths_num_intervals=FLAGS.n_context_length_intervals, |
| 436 | document_depth_percent_intervals=FLAGS.n_document_depth_intervals, |
| 437 | ) |
| 438 | ht.start_test() |
| 439 | |
| 440 | if __name__ == "__main__": |
| 441 | run(main) |
nothing calls this directly
no test coverage detected