mirror of
https://github.com/vale981/ray
synced 2025-03-08 19:41:38 -05:00
31 lines
1.1 KiB
YAML
31 lines
1.1 KiB
YAML
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.dev7-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."
|