From c8895a77800e4ad882deb31963a8f94e457eabc8 Mon Sep 17 00:00:00 2001
From: troyyyyy <huwc@lamda.nju.edu.cn>
Date: Mon, 8 Jan 2024 16:27:21 +0800
Subject: [PATCH] [FIX,DOC] remove mock packages

---
 docs/conf.py          | 27 ---------------------------
 docs/requirements.txt |  8 +++++++-
 2 files changed, 7 insertions(+), 28 deletions(-)

diff --git a/docs/conf.py b/docs/conf.py
index fbb9aa7..5d9deb3 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -10,33 +10,6 @@ from sphinx.application import Sphinx
 # -- 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
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