Try mock imports for Torch to build docs

unreak · Jan 30, 2020 · 023fd73 · 023fd73
1 parent c1a12c4
commit 023fd73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -25,14 +25,15 @@
 sys.path.insert(0, dirname(dirname(__file__)))
 
 # Mock mpi4py to get around having to install it on RTD server (which fails)
+# Also to mock PyTorch, because it is too large for the RTD server to download
 from unittest.mock import MagicMock
 
 class Mock(MagicMock):
     @classmethod
     def __getattr__(cls, name):
         return MagicMock()
 
-MOCK_MODULES = ['mpi4py']
+MOCK_MODULES = ['mpi4py', 'torch']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
 
 # Finish imports

diff --git a/docs/docs_requirements.txt b/docs/docs_requirements.txt
@@ -13,5 +13,4 @@ sphinx==1.5.6
 sphinx-autobuild==0.7.1       
 sphinx-rtd-theme==0.4.1 
 tensorflow>=1.8.0,<2.0
-torch==1.3.1
 tqdm