search

ott-jax / ott

Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
https://ott-jax.readthedocs.io
Apache License 2.0
524 stars 80 forks source link

Feature/update prox #575

Closed michalk8 closed 2 months ago

michalk8 commented 2 months ago

closes #574

review-notebook-app[bot] commented 2 months ago

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

codecov[bot] commented 2 months ago

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 87.78%. Comparing base (ea7d5b9) to head (3c41171). Report is 1 commits behind head on main.

Additional details and impacted files [![Impacted file tree graph](https://app.codecov.io/gh/ott-jax/ott/pull/575/graphs/tree.svg?width=650&height=150&src=pr&token=14PUIHGLV9&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ott-jax)](https://app.codecov.io/gh/ott-jax/ott/pull/575?src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ott-jax) ```diff @@ Coverage Diff @@ ## main #575 +/- ## ========================================== + Coverage 87.76% 87.78% +0.01% ========================================== Files 72 72 Lines 7791 7801 +10 Branches 1124 1126 +2 ========================================== + Hits 6838 6848 +10 Misses 802 802 Partials 151 151 ``` | [Files with missing lines](https://app.codecov.io/gh/ott-jax/ott/pull/575?dropdown=coverage&src=pr&el=tree&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ott-jax) | Coverage Δ | | |---|---|---| | [src/ott/geometry/costs.py](https://app.codecov.io/gh/ott-jax/ott/pull/575?src=pr&el=tree&filepath=src%2Fott%2Fgeometry%2Fcosts.py&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ott-jax#diff-c3JjL290dC9nZW9tZXRyeS9jb3N0cy5weQ==) | `97.22% <100.00%> (+0.05%)` | :arrow_up: | | [src/ott/geometry/regularizers.py](https://app.codecov.io/gh/ott-jax/ott/pull/575?src=pr&el=tree&filepath=src%2Fott%2Fgeometry%2Fregularizers.py&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=ott-jax#diff-c3JjL290dC9nZW9tZXRyeS9yZWd1bGFyaXplcnMucHk=) | `98.65% <100.00%> (+0.01%)` | :arrow_up: |