diff options
-rw-r--r-- | docs/source/conf.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/docs/source/conf.py b/docs/source/conf.py index a099ce0..1304464 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -31,7 +31,8 @@ class Mock(MagicMock): @classmethod def __getattr__(cls, name): return MagicMock() -MOCK_MODULES = ['ot.lp.emd_wrap','autograd','pymanopt'] +MOCK_MODULES = ['ot.lp.emd_wrap','autograd.numpy','pymanopt.manifolds','pymanopt' + 'pymanopt.solvers','cudamat'] sys.modules.update((mod_name, Mock()) for mod_name in MOCK_MODULES) # !!!! |