Skip to content

Commit c3115bc

Browse files
committed
mockj module cupy for gpu focumentation
1 parent 53b063e commit c3115bc

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/source/conf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,9 +36,9 @@ def __getattr__(cls, name):
3636
return MagicMock()
3737

3838

39-
MOCK_MODULES = ['ot.lp.emd_wrap', 'autograd', 'pymanopt', 'cupy', 'autograd.numpy', 'pymanopt.manifolds', 'pymanopt.solvers']
39+
MOCK_MODULES = [ 'cupy']
4040
# 'autograd.numpy','pymanopt.manifolds','pymanopt.solvers',
41-
#sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
41+
sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES)
4242
# !!!!
4343

4444
# If extensions (or modules to document with autodoc) are in another directory,

0 commit comments

Comments
 (0)