From 6ec833a901ad28b1267c8bb458f4b45059c739cf Mon Sep 17 00:00:00 2001
From: Craig Warren <craig_warren@me.com>
Date: Tue, 3 Nov 2015 13:33:52 +0000
Subject: [PATCH] Revised mock imports.

---
 docs/source/conf.py | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

diff --git a/docs/source/conf.py b/docs/source/conf.py
index 62d74cbd..315dfba9 100644
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@@ -20,17 +20,6 @@ import os, re, shlex, sys
 # documentation root, use os.path.abspath to make it absolute, like shown here.
 sys.path.insert(0, os.path.abspath('../..'))
 
-from unittest.mock import MagicMock
-
-class Mock(MagicMock):
-    @classmethod
-    def __getattr__(cls, name):
-        return Mock()
-
-MOCK_MODULES = ['h5py', 'matplotlib']
-sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
-
-
 # -- General configuration ------------------------------------------------
 
 # If your documentation needs a minimal Sphinx version, state it here.
@@ -43,6 +32,9 @@ extensions = [
     'sphinx.ext.mathjax', 'sphinx.ext.autodoc', 'sphinx.ext.napoleon'
 ]
 
+# autodoc mock imports
+autodoc_mock_imports = ['h5py', 'matplotlib']
+
 # Options for autodoc
 autodoc_default_flags = ['members']
 autodoc_member_order = 'bysource'