name: long-running-tests
description: "Ray's long running stress tests"

cluster: ray-project/cluster.yaml

commands:
  - name: run
    help: "Start a long running stress test."
    command: |
      # Install nightly Ray wheels.
      source activate tensorflow_p36 && pip install -U {{wheel}}
      source activate tensorflow_p36 && pip install ray[rllib] ray[debug] gym[atari]
      source activate tensorflow_p36 && pip install ray[debug]
      source activate tensorflow_p36 && python workloads/{{workload}}.py
    params:
      - name: wheel
        help: "URL to the ray wheel to test (defaults to latest)."
        default: https://s3-us-west-2.amazonaws.com/ray-wheels/latest/ray-0.8.0.dev6-cp36-cp36m-manylinux1_x86_64.whl
      - name: workload
        help: "Name of the workload to run."
        choices: ["actor_deaths", "apex", "impala", "many_actor_tasks", "many_drivers", "many_tasks", "node_failures", "pbt"]
    config:
      tmux: true

  - name: check-load
    command: uptime
    help: "Check load of the workload."

  - name: show-output
    command: tmux capture-pane -p
    help: "Show tail of the workoad output."