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

svekars · msaroufim · web-flow · commit b34d1acaa9bd · 2024-05-07T11:06:25.000-07:00
Install dev-requirements.txt

---------

Co-authored-by: Mark Saroufim &lt;marksaroufim@meta.com&gt;
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
