- update conf.py again

inferno-pytorch · Sep 1, 2017 · effc51d · effc51d
1 parent cc631ee
commit effc51d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docs/conf.py b/docs/conf.py
@@ -30,7 +30,8 @@ def __getattr__(cls, name):
                 'pandas',
                 'torch',
                 'torch.nn', 'torch.nn.init', 'torch.nn.functional', 'torch.nn.parallel',
-                'torch.multiprocessing', 'torch.autograd', 'torch.utils',
+                'torch.multiprocessing', 'torch.autograd',
+                'torch.utils', 'torch.utils.data',
                 'torch.optim', 'torch.sparse', 'torch.cuda']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)