group: rllib contrib tests
sort_key: "|rllib_contrib"
depends_on:
  - oss-ci-base_build
steps:
  - block: "run rllib contrib tests"
    if: build.env("BUILDKITE_PIPELINE_ID") == "0189942e-0876-4b8f-80a4-617f988ec59b" && build.branch == "master"

  - label: ":collaborator: rllib contrib: {{matrix}} tests"
    if: build.env("BUILDKITE_PIPELINE_ID") == "0189942e-0876-4b8f-80a4-617f988ec59b"
    instance_type: large
    tags: rllib_contrib
    commands:
      - ./ci/ray_ci/rllib_contrib/rllib_contrib_ci.sh test_{{matrix}}
    job_env: oss-ci-base_build
    matrix:
      - a2c
      - alpha_star
      - alpha_zero
      - apex_ddpg
      - apex_dqn
      - ars
      - bandit
      - ddpg
      - es
      - maddpg
      - maml
      - pg
      - qmix
      - r2d2
      - simple_q
      - slate_q
      - td3

  - label: ":collaborator: rllib contrib: {{matrix}} tests"
    if: build.env("BUILDKITE_PIPELINE_ID") == "0189942e-0876-4b8f-80a4-617f988ec59b"
    soft_fail: true
    instance_type: large
    tags: rllib_contrib
    commands:
      - ./ci/ray_ci/rllib_contrib/rllib_contrib_ci.sh test_{{matrix}}
    job_env: oss-ci-base_build
    matrix:
      - a3c
      - crr
      - ddppo
      - dt
      - leela_chess_zero
      - mbmpo
      - td3