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 -----------------------------------------------------