diff options
Diffstat (limited to 'ot/__init__.py')
-rw-r--r-- | ot/__init__.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index ad7b982..35d2ddd 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -10,7 +10,10 @@ a number of functions described below. :py:mod:`ot.gromov`, :py:mod:`ot.smooth` :py:mod:`ot.stochastic` - The other sub-modules are not imported due to additional dependencies. + The following sub-modules are not imported due to additional dependencies: + + - :any:`ot.dr` : depends on :code:`pymanopt` and :code:`autograd`. + - :any:`ot.gpu` : depends on :code:`cupy` and a CUDA GPU. """ |