From ee8ed4fa101861eec9e578f09aee4367af593af1 Mon Sep 17 00:00:00 2001 From: Rémi Flamary Date: Fri, 28 Sep 2018 09:41:22 +0200 Subject: update documentation --- docs/source/all.rst | 6 ++++++ docs/source/conf.py | 2 +- 2 files changed, 7 insertions(+), 1 deletion(-) (limited to 'docs/source') diff --git a/docs/source/all.rst b/docs/source/all.rst index 9459023..1eaf3b1 100644 --- a/docs/source/all.rst +++ b/docs/source/all.rst @@ -48,6 +48,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) # !!!! -- cgit v1.2.3