diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2021-11-05 17:13:14 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-11-05 17:13:14 +0100 |
| commit | cec41d3817067a2eb3031092735347efe4184237 (patch) | |
| tree | e5af7c2e72fd6f50590b2dd1c5f1f6f47dceebc3 /ot | |
| parent | 0eac835c70cc1a13bb998f3b6cdb0515fafc05e1 (diff) | |
[MRG] Release 0.8 (#289)
* working on release
* test circleci
* try again
* cleanup circle ci run
* add all PR and releant Issues
* update doc
* thanks idris
* update version + add pyproject.toml
* test pyproject.toml
* revert tests
* build wheels
* use windows-latest for tests
* add tests python 3.10
* build all whels
* all versions
* build all wheels
* build all wheels
* cleanup pep8 and minimal acions
* forst shot text release
* bettr text
* stuff
* release text updated
* update manifest to allow build from source
* update doc again
* update release
Diffstat (limited to 'ot')
| -rw-r--r-- | ot/__init__.py | 8 | ||||
| -rw-r--r-- | ot/gpu/__init__.py | 10 | ||||
| -rw-r--r-- | ot/helpers/__init__.py | 3 |
3 files changed, 16 insertions, 5 deletions
diff --git a/ot/__init__.py b/ot/__init__.py index 4292b41..b6dc2b4 100644 --- a/ot/__init__.py +++ b/ot/__init__.py @@ -5,7 +5,8 @@ :py:mod:`ot.lp`, :py:mod:`ot.bregman`, :py:mod:`ot.optim` :py:mod:`ot.utils`, :py:mod:`ot.datasets`, :py:mod:`ot.gromov`, :py:mod:`ot.smooth` - :py:mod:`ot.stochastic` + :py:mod:`ot.stochastic`, :py:mod:`ot.partial`, :py:mod:`ot.regpath` + , :py:mod:`ot.unbalanced`. The following sub-modules are not imported due to additional dependencies: @@ -49,7 +50,7 @@ from .gromov import (gromov_wasserstein, gromov_wasserstein2, # utils functions from .utils import dist, unif, tic, toc, toq -__version__ = "0.8.0dev" +__version__ = "0.8.0" __all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', 'datasets', 'bregman', 'lp', 'tic', 'toc', 'toq', 'gromov', @@ -57,5 +58,6 @@ __all__ = ['emd', 'emd2', 'emd_1d', 'sinkhorn', 'sinkhorn2', 'utils', 'dist', 'unif', 'barycenter', 'sinkhorn_lpl1_mm', 'da', 'optim', 'sinkhorn_unbalanced', 'barycenter_unbalanced', 'sinkhorn_unbalanced2', 'sliced_wasserstein_distance', - 'max_sliced_wasserstein_distance', + 'gromov_wasserstein', 'gromov_wasserstein2', 'gromov_barycenters', 'fused_gromov_wasserstein', 'fused_gromov_wasserstein2', + 'max_sliced_wasserstein_distance', 'smooth', 'stochastic', 'unbalanced', 'partial', 'regpath'] diff --git a/ot/gpu/__init__.py b/ot/gpu/__init__.py index e939610..12db605 100644 --- a/ot/gpu/__init__.py +++ b/ot/gpu/__init__.py @@ -7,7 +7,13 @@ The GPU backend in handled by `cupy <https://cupy.chainer.org/>`_. .. warning:: - Note that by default the module is not import in :mod:`ot`. In order to + This module is now deprecated and will be removed in future releases. POT + now privides a backend mechanism that allows for solving prolem on GPU wth + the pytorch backend. + + +.. warning:: + Note that by default the module is not imported in :mod:`ot`. In order to use it you need to explicitely import :mod:`ot.gpu` . By default, the functions in this module accept and return numpy arrays @@ -36,7 +42,7 @@ from . import utils from .utils import dist, to_gpu, to_np -warnings.warn('This module will be deprecated in the next minor release of POT', category=DeprecationWarning) +warnings.warn('This module is deprecated and will be removed in the next minor release of POT', category=DeprecationWarning) __all__ = ["utils", "dist", "sinkhorn", diff --git a/ot/helpers/__init__.py b/ot/helpers/__init__.py new file mode 100644 index 0000000..b948671 --- /dev/null +++ b/ot/helpers/__init__.py @@ -0,0 +1,3 @@ +# Author: Remi Flamary <remi.flamary@unice.fr> +# +# License: MIT License |