def run_vllm_nightly_on_v6e_4_persistent_TPUVM()

in dags/mantaray/run_mantaray_jobs.py [0:0]

• 18 lines of code
• 1 McCabe index (conditional complexity)

  def run_vllm_nightly_on_v6e_4_persistent_TPUVM(current_request_rate: int):
    ### output_location path get to save metric_result.jsonl
    output_location = get_path(current_request_rate)
    print("output_location: ", output_location)

    ### run test code
    run_on_v6e_4_persistent_TPUVM_func = run_on_v6e_4_persistent_TPUVM(
        output_location, current_request_rate
    )

    ### push test result to BigQuery Database
    ### task_metric_config: generate metric configs to push test result
    job_metric_config = metric_config.MetricConfig(
        json_lines="metric_report.jsonl",
        use_runtime_generated_gcs_folder=True,
    )

    process_id = metric.generate_process_id.override(retries=0)()
    print("process_id: ", process_id)
    post_process = new_process_metrics(
        str(process_id),
        job_metric_config,
        folder_location=output_location,
    )

    ### order of execute
    run_on_v6e_4_persistent_TPUVM_func >> post_process