diff --git a/.github/workflows/codeqc.yml b/.github/workflows/codeqc.yml
index 23f5dfc..e132eb8 100644
--- a/.github/workflows/codeqc.yml
+++ b/.github/workflows/codeqc.yml
@@ -53,4 +53,18 @@ jobs:
       run: |
         conda activate torchsurv
         ./dev/run-doctests.sh
-        ./dev/run-unittests.sh
\ No newline at end of file
+        ./dev/run-unittests.sh
+
+    - name: Test-build docs
+      shell: bash -l {0}
+      run: |
+        conda activate torchsurv
+        ./dev/build-docs.sh
+        tar czvf docs.tar.gz -C docs/_build/html .
+
+    - name: 'Upload Docs Tarball'
+      uses: actions/upload-artifact@v4
+      with:
+        name: docs.tar.gz
+        path: docs.tar.gz
+        retention-days: 5
\ No newline at end of file
diff --git a/docs/conf.py b/docs/conf.py
index 3257ded..350f69c 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -26,6 +26,7 @@
     "sphinx.ext.coverage",
     "sphinx.ext.napoleon",
     "nbsphinx",
+    "sphinx.ext.viewcode",
     "sphinxcontrib.bibtex",
 ]