def get_vllm_tpu_setup_cmds()

in dags/solutions_team/configs/vllm/vllm_benchmark_config.py [0:0]

• 13 lines of code
• 3 McCabe index (conditional complexity)

def get_vllm_tpu_setup_cmds():
  setup_cmds = (
      # Download and start the vLLM TPU Docker container
      f"export CONTAINER_NAME={VLLM_TPU_CONTAINER}",
      f"sudo docker run --name $CONTAINER_NAME -d --privileged --network host -v /dev/shm:/dev/shm {VLLM_TPU_DOCKER_IMAGE} tail -f /dev/null",
      # Download dataset inside the container
      "sudo docker exec $CONTAINER_NAME /bin/bash -c 'wget --no-verbose https://huggingface.co/datasets/anon8231489123/ShareGPT_Vicuna_unfiltered/resolve/main/ShareGPT_V3_unfiltered_cleaned_split.json'",
      # Download benchmark inside the container
      "sudo docker exec $CONTAINER_NAME /bin/bash -c 'pip install --upgrade google-cloud-storage'",
      "sudo docker exec $CONTAINER_NAME /bin/bash -c 'rm -rf inference-benchmark && git clone https://github.com/AI-Hypercomputer/inference-benchmark'",
      # Download Google Cloud SDK inside the container, which is needed for the gsutil command.
      "sudo docker exec $CONTAINER_NAME /bin/bash -c 'echo \"deb [signed-by=/usr/share/keyrings/cloud.google.gpg] http://packages.cloud.google.com/apt cloud-sdk main\" > /etc/apt/sources.list.d/google-cloud-sdk.list'",
      "sudo docker exec $CONTAINER_NAME /bin/bash -c 'curl https://packages.cloud.google.com/apt/doc/apt-key.gpg | apt-key --keyring /usr/share/keyrings/cloud.google.gpg add -'",
      "sudo docker exec $CONTAINER_NAME /bin/bash -c 'apt-get update && apt-get install -y google-cloud-sdk'",
      "sudo docker exec $CONTAINER_NAME /bin/bash -c 'apt-get -y install jq'",
  )

  return setup_cmds