(argv)
| 434 | |
| 435 | |
| 436 | def main(argv): |
| 437 | JaxDistributedConfig.initialize(FLAGS.jax_distributed) |
| 438 | set_random_seed(FLAGS.seed) |
| 439 | |
| 440 | ht = LLMNeedleHaystackTester( |
| 441 | haystack_file=FLAGS.haystack_file, |
| 442 | context_lengths_min=FLAGS.context_lengths_min, |
| 443 | context_lengths_max=FLAGS.context_lengths_max, |
| 444 | context_lengths_num_intervals=FLAGS.n_context_length_intervals, |
| 445 | document_depth_percent_intervals=FLAGS.n_document_depth_intervals, |
| 446 | ) |
| 447 | ht.start_test() |
| 448 | |
| 449 | if __name__ == "__main__": |
| 450 | run(main) |
nothing calls this directly
no test coverage detected