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

---
 docs/conf.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/docs/conf.py b/docs/conf.py
index 44ffd72..fbb9aa7 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -31,6 +31,8 @@ MOCK_MODULES = [
     "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)