load torch into docs

liopeer · Sep 2, 2023 · 4b78bea · 4b78bea
1 parent 359c4aa
commit 4b78bea
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -5,4 +5,7 @@ myst_parser
 sphinx-autodoc-typehints
 jaxtyping
 pandoc
+torch 
+torchvision 
+torchaudio
 git+https://github.com/liopeer/type_hints.git
diff --git a/docs/source/_static b/docs/source/_static