Add mock numpy and scipy modules to docs/conf.py

albumentations-team · Jun 19, 2018 · a613d5b · a613d5b
1 parent 7115799
commit a613d5b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -1,11 +1,17 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
+import mock
 import re
 import os
 import sys
 sys.path.insert(0, os.path.abspath('../'))
 
 
+MOCK_MODULES = ['numpy', 'scipy']
+for module_name in MOCK_MODULES:
+    sys.modules[module_name] = mock.Mock()
+
+
 def get_version():
     current_dir = os.path.abspath(os.path.dirname(__file__))
     root = os.path.dirname(current_dir)