build-docs job builds from source (#2466)

Summary:
Pull Request resolved: #2466

Previously we build docs from nightly. Hard for us to see previews of docstring changes, here we change it to build from source.

Reviewed By: svekars

Differential Revision: D63784555

fbshipit-source-id: 2402fc7d31b91c9f1b7a503211c9d3e6b8f7641e

.github/workflows/docs.yml

@@ -18,6 +18,7 @@ jobs:
         include:
          - os: linux.20_04.4x
            python-version: 3.9
+           python-tag: "py39"
     steps:
     - name: Check ldd --version
       run: ldd --version
@@ -66,7 +67,8 @@ jobs:
         conda run -n build_binary pip install torchmetrics==1.0.3
     - name: Install TorchRec
       run: |
-        conda run -n build_binary pip install torchrec --index-url https://download.pytorch.org/whl/nightly/cpu
+        conda run -n build_binary pip install -r requirements.txt
+        conda run -n build_binary python setup.py bdist_wheel --python-tag=${{ matrix.python-tag }}
     - name: Test fbgemm_gpu and torchrec installation
       shell: bash
       run: |

docs/source/conf.py

@@ -21,7 +21,6 @@
 import sys
 
 import pytorch_sphinx_theme
-import torchrec
 
 current_dir = os.path.dirname(__file__)
 target_dir = os.path.abspath(os.path.join(current_dir, "../.."))
@@ -36,7 +35,7 @@
 
 try:
     # pyre-ignore
-    version = torchrec.__version__
+    version = "1.0.0"  # TODO: Hardcode stable version for now
 except Exception:
     # when run internally, we don't have a version yet
     version = "0.0.0"

torchrec/distributed/planner/planners.py

@@ -156,6 +156,7 @@ class EmbeddingShardingPlanner(ShardingPlanner):
         debug (bool): whether to print debug information.
 
     Example::
+
         ebc = EmbeddingBagCollection(tables=eb_configs, device=torch.device("meta"))
         planner = EmbeddingShardingPlanner()
         plan = planner.plan(