diff --git a/docs/conf.py b/docs/conf.py
index 3c3df49..326b4ce 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -12,17 +12,6 @@ from sphinx.application import Sphinx
 sys.path.insert(0, os.path.abspath(".."))
 
 import ablkit  # noqa: E402
-from unittest.mock import MagicMock
-
-
-class Mock(MagicMock):
-    @classmethod
-    def __getattr__(cls, name):
-        return MagicMock()
-
-
-MOCK_MODULES = ["numpy", "pyswip", "torch", "torchvision", "zoopt", "termcolor"]
-sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
 
 # -- Project information -----------------------------------------------------
 
diff --git a/docs/requirements.txt b/docs/requirements.txt
index 88bedfb..ebde2fe 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -2,4 +2,10 @@ sphinx
 sphinx-rtd-theme 
 recommonmark
 sphinx-markdown-tables
-sphinx-copybutton
\ No newline at end of file
+sphinx-copybutton
+numpy
+pyswip
+torch
+torchvision
+zoopt
+termcolor
\ No newline at end of file