readd pytorch dependency

cpnota · Jan 14, 2020 · 88c0049 · 88c0049
1 parent 22576b1
commit 88c0049
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/.readthedocs.yml b/.readthedocs.yml
@@ -24,3 +24,4 @@ python:
       path: .
       extra_requirements:
         - docs
+        - pytorch
diff --git a/docs/source/conf.py b/docs/source/conf.py
@@ -42,12 +42,6 @@
 autodoc_default_flags = ['members']
 autosummary_generate = True
 
-# Mock requirements to save resources during doc build machine setup
-autodoc_mock_imports = [
-    'torch',
-    'torchvision',
-]
-
 # Add any paths that contain templates here, relative to this directory.
 templates_path = ['_templates']