[FIX,DOC] remove mock packages

AbductiveLearning · Jan 8, 2024 · c8895a7 · c8895a7
1 parent ad275d6
commit c8895a7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 28 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -10,33 +10,6 @@
 # -- Path setup --------------------------------------------------------------
 
 sys.path.insert(0, os.path.abspath(".."))
-
-from unittest.mock import MagicMock
-
-
-class Mock(MagicMock):
-    @classmethod
-    def __getattr__(cls, name):
-        return MagicMock()
-
-
-MOCK_MODULES = [
-    "numpy",
-    "pyswip",
-    "torch",
-    "torchvision",
-    "zoopt",
-    "termcolor",
-    "torch.nn",
-    "torch.utils",
-    "torch.optim",
-    "torchvision.transform",
-    "torch.utils.data",
-    "torch.utils.data.sampler"
-]
-sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
-
-
 import ablkit  # noqa: E402
 
 # -- Project information -----------------------------------------------------

diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -2,4 +2,10 @@ sphinx
 sphinx-rtd-theme 
 recommonmark
 sphinx-markdown-tables
-sphinx-copybutton
+sphinx-copybutton
+numpy
+pyswip
+torch
+torchvision
+zoopt
+termcolor