From 67bd489af64a47588fced2cd41cf6957fa3464e3 Mon Sep 17 00:00:00 2001
From: troyyyyy <huwc@lamda.nju.edu.cn>
Date: Mon, 8 Jan 2024 16:07:09 +0800
Subject: [PATCH] [DOC] remove surplus requirements (cont.)

---
 docs/conf.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/docs/conf.py b/docs/conf.py
index 311f774..44ffd72 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -20,7 +20,18 @@ class Mock(MagicMock):
         return MagicMock()
 
 
-MOCK_MODULES = ["numpy", "pyswip", "torch", "torchvision", "zoopt", "termcolor"]
+MOCK_MODULES = [
+    "numpy",
+    "pyswip",
+    "torch",
+    "torchvision",
+    "zoopt",
+    "termcolor",
+    "torch.nn",
+    "torch.utils",
+    "torch.optim",
+    "torchvision.transform",
+]
 sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)