Merge pull request #839 from azavea/lf/fixdocs-backport

Mock torch-related dependencies in docs
azavea · Oct 2, 2019 · a8a15f2 · a8a15f2
2 parents f08f931 + 8f2b10e
commit a8a15f2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -59,6 +59,8 @@ def __getattr__(cls, name):
 MOCK_MODULES = ['pyproj', 'h5py']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
 
+autodoc_mock_imports = ['torch', 'torchvision', 'pycocotools']
+
 intersphinx_mapping = {'python': ('https://docs.python.org/3/', None)}
 
 # Add any paths that contain templates here, relative to this directory.