diff options
Diffstat (limited to 'docs/source')
| -rw-r--r-- | docs/source/all.rst | 6 | ||||
| -rw-r--r-- | docs/source/conf.py | 2 |
2 files changed, 7 insertions, 1 deletions
diff --git a/docs/source/all.rst b/docs/source/all.rst index 94da2ed..32930fd 100644 --- a/docs/source/all.rst +++ b/docs/source/all.rst @@ -43,6 +43,12 @@ ot.da .. automodule:: ot.da :members: + +ot.gpu +-------- + +.. automodule:: ot.gpu + :members: ot.dr -------- diff --git a/docs/source/conf.py b/docs/source/conf.py index 114245d..433eca6 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -31,7 +31,7 @@ class Mock(MagicMock): @classmethod def __getattr__(cls, name): return MagicMock() -MOCK_MODULES = ['ot.lp.emd_wrap','autograd','pymanopt','cudamat','autograd.numpy','pymanopt.manifolds','pymanopt.solvers'] +MOCK_MODULES = ['ot.lp.emd_wrap','autograd','pymanopt','cupy','autograd.numpy','pymanopt.manifolds','pymanopt.solvers'] # 'autograd.numpy','pymanopt.manifolds','pymanopt.solvers', sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES) # !!!! |