Update conf.py

AxeldeRomblay · Jul 5, 2017 · 3326499 · 3326499
1 parent 906a98b
commit 3326499
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -16,12 +16,12 @@
 import sys
 import os
 
-from unittest.mock import MagicMock
+from mock import Mock as MagicMock
 
 class Mock(MagicMock):
     @classmethod
     def __getattr__(cls, name):
-            return MagicMock()
+        return MagicMock()
 
 MOCK_MODULES = ['lightgbm', 'xgboost','numpy', 'pandas']
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)