- name: dask_xgboost_test
  team: ml
  cluster:
    app_config: dask_xgboost_app_config.yaml
    compute_template: compute_tpl.yaml

  run:
    use_connect: True
    autosuspend_mins: 10
    timeout: 1200
    script: python workloads/dask_xgboost_test.py
    args:
      [
        "--num-actors 4",
        "--cpus-per-actor 4",
        "--num-actors-inference 16",
        "--cpus-per-actor-inference 1",
      ]

- name: modin_xgboost_test
  team: ml
  cluster:
    app_config: modin_xgboost_app_config.yaml
    compute_template: compute_tpl.yaml

  run:
    use_connect: True
    autosuspend_mins: 10
    timeout: 1200
    script: python workloads/modin_xgboost_test.py
    args:
      [
        "--num-actors 4",
        "--cpus-per-actor 4",
        "--num-actors-inference 16",
        "--cpus-per-actor-inference 1",
      ]

- name: torch_tune_serve_test
  team: ml
  cluster:
    app_config: torch_tune_serve_app_config.yaml
    compute_template: gpu_tpl.yaml

  run:
    use_connect: True
    autosuspend_mins: 10
    timeout: 1200
    script: python workloads/torch_tune_serve_test.py