- update conf.py

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