pytorch · kiukchung · Apr 26, 2021
diff --git a/docs/Makefile b/docs/Makefile
@@ -10,7 +10,7 @@ SPHINXBUILD   = sphinx-build
 SPHINXPROJ    = torchelastic
 SOURCEDIR     = source
 BUILDDIR      = build
-VERSION       := $(shell python -c "import torchelastic; print(torchelastic.__version__)")
+VERSION       := "0.2.3.dev0"
 
 # Put it first so that "make" without argument is like "make help".
 help:

diff --git a/docs/source/agent.rst b/docs/source/agent.rst
diff --git a/docs/source/agent_diagram.jpg b/docs/source/agent_diagram.jpg
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -22,8 +22,6 @@
 # import os
 # import sys
 # sys.path.insert(0, os.path.abspath('.'))
-import torch
-import torchelastic
 from docutils import nodes
 from sphinx import addnodes
 from sphinx.util.docfields import TypedField
@@ -88,7 +86,7 @@
 #
 # The short X.Y version.
 # TODO: change to [:2] at v1.0
-version = f"v{torchelastic.__version__}"
+version = "v0.2.3.dev0"
 # The full version, including alpha/beta/rc tags.
 # TODO: verify this works as expected
 release = "master"

diff --git a/docs/source/customization.rst b/docs/source/customization.rst
diff --git a/docs/source/distributed.rst b/docs/source/distributed.rst
diff --git a/docs/source/errors.rst b/docs/source/errors.rst
diff --git a/docs/source/etcd_rdzv_diagram.png b/docs/source/etcd_rdzv_diagram.png
diff --git a/docs/source/events.rst b/docs/source/events.rst
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -3,49 +3,14 @@
 TorchElastic
 ==================
 
-Makes distributed PyTorch fault-tolerant and elastic.
-
-Get Started
----------------
-.. toctree::
-   :maxdepth: 1
-   :caption: Usage
-
-   quickstart
-   train_script
-   examples
-
-Documentation
----------------
-
-.. toctree::
-   :maxdepth: 1
-   :caption: API
-
-   distributed
-   agent
-   multiprocessing
-   errors
-   rendezvous
-   timer
-   metrics
-   events
-
-.. toctree::
-   :maxdepth: 1
-   :caption: Advanced
-
-   customization
-
-.. toctree::
-   :maxdepth: 1
-   :caption: Plugins
-
-   kubernetes
-   runtime
-
+.. important:: TorchElastic has been upstreamed to PyTorch (``torch.distributed.elastic``) and will be
+               included as part of the next PyTorch release (``torch-1.9``). In the meanwhile please use
+               `pytorch-nightly <https://pytorch.org/get-started/locally/#start-locally>`_ or
+               `torchelastic-0.2.2 <https://pypi.org/project/torchelastic/>`_. For documentation please refer to our
+               `v0.2.2 docs <https://pytorch.org/elastic/0.2.2>`_ for now.
+
 Training Session Manager(TSM)
----------------
+------------------------------
 .. toctree::
    :maxdepth: 1
    :caption: Training Session Manager(TSM)

diff --git a/docs/source/kubernetes.rst b/docs/source/kubernetes.rst
diff --git a/docs/source/metrics.rst b/docs/source/metrics.rst
diff --git a/docs/source/multiprocessing.rst b/docs/source/multiprocessing.rst