Install dev-requirements.txt in doc build (pytorch#224)

Install dev-requirements.txt --------- Co-authored-by: Mark Saroufim <[email protected]>
dbyoung18 · May 7, 2024 · 4b12b6b · 4b12b6b
1 parent ac14b4b
commit 4b12b6b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/.github/workflows/doc_build.yml b/.github/workflows/doc_build.yml
@@ -41,6 +41,7 @@ jobs:
         run: |
           python -m pip install torch
           python -m pip install -e .
+          pip install -r dev-requirements.txt
           cd docs
           python -m pip install -r requirements.txt
       - name: Build docs

diff --git a/docs/source/api_ref_sparsity.rst b/docs/source/api_ref_sparsity.rst
@@ -3,7 +3,7 @@
 ================
 torchao.sparsity
 ================
-
+.. automodule:: torchao.sparsity
 .. currentmodule:: torchao.sparsity
 
 .. autosummary::
@@ -12,3 +12,7 @@ torchao.sparsity
 
     WandaSparsifier
     PerChannelNormObserver
+    apply_sparse_semi_structured
+    apply_fake_sparsity
+
+
diff --git a/docs/source/index.rst b/docs/source/index.rst
@@ -92,9 +92,9 @@ with more content coming soon.
    :maxdepth: 1
    :caption: API Reference
 
-   api_ref_intro
    api_ref_sparsity
+   api_ref_intro
    api_ref_quantization
    api_ref_dtypes
-   ..
+..
       api_ref_kernel