Browse Source

[DOC] remove surplus requirements (cont.)

pull/6/head
troyyyyy 1 year ago
parent
commit
67bd489af6
1 changed files with 12 additions and 1 deletions
  1. +12
    -1
      docs/conf.py

+ 12
- 1
docs/conf.py View File

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




Loading…
Cancel
Save