- added torch submodules to be mocked

inferno-pytorch · Sep 1, 2017 · 84dd528 · 84dd528
1 parent 810a4c6
commit 84dd528
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -23,7 +23,9 @@ class Mock(MagicMock):
     def __getattr__(cls, name):
             return MagicMock()
 
-MOCK_MODULES = ['pygtk', 'hdf5', 'skimage', 'argparse', 'pandas','torch','torch.nn']
+MOCK_MODULES = ['pygtk', 'hdf5', 'skimage', 'argparse', 'pandas', 'torch', 'torch.nn',
+                'torch.multiprocessing', 'torch.autograd', 'torch.optim', 'torch.sparse',
+                'torch.cuda']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)