(torchx/schedulers) Retire ray scheduler

Differential Revision: D82584920

Pull Request resolved: #1120

Author: kiukchung

.github/workflows/components-integration-tests.yaml

@@ -24,8 +24,6 @@ jobs:
             platform: ubuntu-latest
           - scheduler: "local_docker"
             platform: linux.24_04.4x
-          - scheduler: "ray"
-            platform: ubuntu-latest
       fail-fast: false
     runs-on: ${{ matrix.platform }}
     permissions:

README.md

@@ -19,7 +19,6 @@ TorchX currently supports:
 * AWS Batch
 * Docker
 * Local
-* Ray (prototype)
 * GCP Batch (prototype)
 
 Need a scheduler not listed? [Let us know!](https://github.com/pytorch/torchx/issues?q=is%3Aopen+is%3Aissue+label%3Ascheduler-request)
@@ -58,9 +57,6 @@ pip install "torchx[dev]"
 # install torchx Kubernetes / Volcano support
 pip install "torchx[kubernetes]"
 
-# install torchx Ray support
-pip install "torchx[ray]"
-
 # install torchx GCP Batch support
 pip install "torchx[gcp_batch]"
 ```

dev-requirements.txt

@@ -29,7 +29,6 @@ torchtext==0.18.0
 torchvision==0.23.0
 typing-extensions
 ts==0.5.1
-ray[default]
 wheel
 
 # lint (linter versions are managed by lintrunner)

docs/source/index.rst

@@ -74,7 +74,6 @@ Works With
    schedulers/kubernetes
    schedulers/kubernetes_mcad
    schedulers/slurm
-   schedulers/ray
    schedulers/aws_batch
    schedulers/aws_sagemaker
    schedulers/lsf

docs/source/quickstart.md

@@ -174,7 +174,6 @@ run command for local works out of the box on remote.
 $ torchx run --scheduler slurm dist.ddp -j 2x2 --script dist_app.py
 $ torchx run --scheduler kubernetes dist.ddp -j 2x2 --script dist_app.py
 $ torchx run --scheduler aws_batch dist.ddp -j 2x2 --script dist_app.py
-$ torchx run --scheduler ray dist.ddp -j 2x2 --script dist_app.py
 ```
 <!-- #endmd -->
 

docs/source/schedulers/ray.rst

@@ -88,7 +88,6 @@ def get_nightly_version():
                 "google-cloud-runtimeconfig>=0.33.2",
             ],
             "kubernetes": ["kubernetes>=11"],
-            "ray": ["ray>=1.12.1"],
             "dev": dev_reqs,
         },
         # PyPI package information.

setup.py

@@ -22,7 +22,6 @@
     "aws_batch": "torchx.schedulers.aws_batch_scheduler",
     "aws_sagemaker": "torchx.schedulers.aws_sagemaker_scheduler",
     "gcp_batch": "torchx.schedulers.gcp_batch_scheduler",
-    "ray": "torchx.schedulers.ray_scheduler",
     "lsf": "torchx.schedulers.lsf_scheduler",
 }