Fix/docs (#531)

* Fix docs build * Fix docs build
pytorchbearer · Mar 20, 2019 · 513f4a9 · 513f4a9
1 parent 5220a2b
commit 513f4a9
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -19,12 +19,12 @@
 #
 import os
 import sys
-from unittest.mock import MagicMock
 
 autodoc_mock_imports = ['torch', 'torch.nn.utils.clip_grad_norm', 'torchvision', 'torchvision.utils', 'torchvision.datasets', 'torchvision.datasets.folder', 'torch.nn', 'torch.nn.functional', 'torch.nn.modules', 'torch.optim', 'torch.distributions.utils', 'torch.distributions', 'torch.utils', 'torch.utils.data', 'numpy', 'sklearn', 'sklearn.metrics', 'tqdm', 'tensorboardX', 'tensorboardX.torchvis', 'livelossplot', 'IPython']
 
 sys.path.insert(0, os.path.abspath('.'))
 sys.path.insert(0, os.path.abspath('../'))
+sys.path.insert(0, os.path.abspath('../torchbearer'))
 
 
 # -- General configuration ------------------------------------------------

diff --git a/torchbearer/__init__.py b/torchbearer/__init__.py
@@ -6,13 +6,6 @@
         :members:
         :undoc-members:
 
-Model (Deprecated)
-^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
-
-..  automodule:: torchbearer.torchbearer
-        :members:
-        :undoc-members:
-
 State
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^