From bff0b65b4f78ba77157a74a690132ca3e2dc60f2 Mon Sep 17 00:00:00 2001 From: troyyyyy Date: Mon, 8 Jan 2024 15:49:31 +0800 Subject: [PATCH] [DOC] add autodoc mock --- docs/conf.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/docs/conf.py b/docs/conf.py index 326b4ce..3c3df49 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -12,6 +12,17 @@ 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 -----------------------------------------------------