# Todo: Enable once tests are available
#- label: ":tv: :octopus: Tune GPU tests "
#  conditions: ["NO_WHEELS_REQUIRED", "RAY_CI_TUNE_AFFECTED"]
#  commands:
#    - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/build/upload_build_info.sh; fi }; trap cleanup EXIT
#    - TUNE_TESTING=1 ./ci/env/install-dependencies.sh
#    - pip install -Ur ./python/requirements/ml/requirements_ml_docker.txt
#    - ./ci/env/env_info.sh
#    - bazel test --config=ci $(./ci/run/bazel_export_options) --build_tests_only --test_tag_filters=gpu python/ray/tune/...

- label: ":tv: :brain: RLlib: GPU Tests"
  conditions: ["NO_WHEELS_REQUIRED", "RAY_CI_RLLIB_AFFECTED"]
  commands:
    - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/build/upload_build_info.sh; fi }; trap cleanup EXIT
    - RLLIB_TESTING=1 ./ci/env/install-dependencies.sh
    - pip install -Ur ./python/requirements/ml/requirements_ml_docker.txt
    - ./ci/env/env_info.sh
    # --jobs 1 is necessary as we only have 1 GPU on the machine and running tests in parallel
    # would cause timeouts as the other scripts would wait for the GPU to become available.
    - bazel test --config=ci $(./ci/run/bazel_export_options) --build_tests_only --jobs 1
      --test_tag_filters=gpu --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1 rllib/...


- label: ":tv: :serverless: Serve Tests"
  conditions:
    [
        "NO_WHEELS_REQUIRED",
        "RAY_CI_SERVE_AFFECTED",
        "RAY_CI_PYTHON_AFFECTED",
        "RAY_CI_ML_AFFECTED",
    ]
  commands:
    - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/build/upload_build_info.sh; fi }; trap cleanup EXIT
    - ./ci/env/install-dependencies.sh
    - pip install -Ur ./python/requirements/ml/requirements_ml_docker.txt
    - bazel test --config=ci $(./ci/run/bazel_export_options) --test_tag_filters=gpu python/ray/serve/...

# Todo: enable once tests pass
#- label: ":tv: :brain: RLlib: GPU Examples {C/D}"
#  conditions: ["NO_WHEELS_REQUIRED", "RAY_CI_RLLIB_AFFECTED"]
#  commands:
#    - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/build/upload_build_info.sh; fi }; trap cleanup EXIT
#    - RLLIB_TESTING=1 ./ci/env/install-dependencies.sh
#    - pip install -Ur ./python/requirements/ml/requirements_ml_docker.txt
#    - ./ci/env/env_info.sh
#    - bazel test --config=ci $(./ci/run/bazel_export_options) --build_tests_only
#      --test_tag_filters=examples_C,examples_D --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1 rllib/...

# Todo: enable once tests pass
#- label: ":tv: :brain: RLlib: GPU Examples {E/P}"
#  conditions: ["NO_WHEELS_REQUIRED", "RAY_CI_RLLIB_AFFECTED"]
#  commands:
#    - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/build/upload_build_info.sh; fi }; trap cleanup EXIT
#    - RLLIB_TESTING=1 ./ci/env/install-dependencies.sh
#    - pip install -Ur ./python/requirements/ml/requirements_ml_docker.txt
#    - ./ci/env/env_info.sh
#    - bazel test --config=ci $(./ci/run/bazel_export_options) --build_tests_only
#      --test_tag_filters=examples_E,examples_F,examples_G,examples_H,examples_I,examples_J,examples_K,examples_L,examples_M,examples_N,examples_O,examples_P --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1
#      rllib/...

# Todo: enable once tests pass
#- label: ":tv: :brain: RLlib: GPU Examples {Q/Z}"
#  conditions: ["NO_WHEELS_REQUIRED", "RAY_CI_RLLIB_AFFECTED"]
#  commands:
#    - cleanup() { if [ "${BUILDKITE_PULL_REQUEST}" = "false" ]; then ./ci/build/upload_build_info.sh; fi }; trap cleanup EXIT
#    - RLLIB_TESTING=1 ./ci/env/install-dependencies.sh
#    - pip install -Ur ./python/requirements/ml/requirements_ml_docker.txt
#    - ./ci/env/env_info.sh
#    - bazel test --config=ci $(./ci/run/bazel_export_options) --build_tests_only
#      --test_tag_filters=examples_Q,examples_R,examples_S,examples_T,examples_U,examples_V,examples_W,examples_X,examples_Y,examples_Z --test_env=RAY_USE_MULTIPROCESSING_CPU_COUNT=1 --test_env=RLLIB_NUM_GPUS=1
#      rllib/...