awesome mini gallery

Author: rflamary

docs/source/all.rst

@@ -4,96 +4,33 @@
 API and modules
 ===============
 
-ot
---
+.. currentmodule:: ot
 
-.. automodule:: ot
-   :members:
-
-ot.lp
------
-.. automodule:: ot.lp
-   :members:
-
-ot.bregman
-----------
-
-.. automodule:: ot.bregman
-   :members:
-
-ot.smooth
------
-.. automodule:: ot.smooth
-   :members:
-
-ot.gromov
-----------
-
-.. automodule:: ot.gromov
-   :members:
-
-
-ot.optim
---------
-
-.. automodule:: ot.optim
-   :members:
-
-ot.da
---------
-
-.. automodule:: ot.da
-  :members:
-
-ot.gpu
---------
-
-.. automodule:: ot.gpu
-  :members:
 
-ot.dr
---------
+:py:mod:`ot`:
 
-.. automodule:: ot.dr
-  :members:
-
-
-ot.utils
---------
-
-.. automodule:: ot.utils
-   :members:
-
-ot.datasets
------------
-
-.. automodule:: ot.datasets
-   :members:
-
-ot.plot
--------
-
-.. automodule:: ot.plot
-   :members:
+.. autosummary::
+   :toctree: gen_modules/
+   :template: module.rst
+
+   lp
+   bregman
+   smooth
+   gromov
+   optim
+   da
+   gpu
+   dr
+   utils
+   datasets
+   plot
+   stochastic
+   unbalanced
+   partial
 
-ot.stochastic
--------------
+.. autosummary::
+   :toctree: ../modules/generated/
+   :template: module.rst
 
-.. automodule:: ot.stochastic
+.. automodule:: ot
    :members:
-
-ot.unbalanced
--------------
-
-.. automodule:: ot.unbalanced
-  :members:
-  
-ot.partial
--------------
-
-.. automodule:: ot.partial
-  :members:
-
-.. autosummary::
-   :toctree: gen_modules/
-   :template: module.rst

docs/source/conf.py

@@ -71,6 +71,9 @@ def __getattr__(cls, name):
     'sphinx_gallery.gen_gallery',
 ]
 
+autosummary_generate = True
+
+
 napoleon_numpy_docstring = True
 
 # Add any paths that contain templates here, relative to this directory.