Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

update mm-sink tutorial and plot #572

Merged
merged 10 commits into from
Aug 21, 2024
Merged

Conversation

zoepiran
Copy link
Contributor

Updates MM-OT tutorial visualization and adds MM-OT plotting utils.

Updates in files:

  • docs/references.bib: add citation lin:22
  • docs/tutorials/linear/600_mmsink.ipynb: update visuals and add comparison to MM (det. solution)
  • src/ott/tools/plot.py: add class PlotMM

Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

docs/references.bib Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
src/ott/tools/plot.py Outdated Show resolved Hide resolved
Copy link

review-notebook-app bot commented Aug 19, 2024

View / edit / reply to this conversation on ReviewNB

michalk8 commented on 2024-08-19T10:41:15Z
----------------------------------------------------------------

Line #1.    def plot_clouds(

With a bit of refactoring, I think we should be able to do

plott = plot.PlotMM(fig=fig, cmap=cmap)
_ = plott(out)

And then we could remove the helper plotting function above.


zoepiran commented on 2024-08-19T14:29:06Z
----------------------------------------------------------------

genious

Copy link

review-notebook-app bot commented Aug 19, 2024

View / edit / reply to this conversation on ReviewNB

michalk8 commented on 2024-08-19T10:41:15Z
----------------------------------------------------------------

Both are determisnistic, afaik.


zoepiran commented on 2024-08-19T14:29:19Z
----------------------------------------------------------------

horrible phrasing

Copy link

codecov bot commented Aug 19, 2024

Codecov Report

Attention: Patch coverage is 18.84058% with 56 lines in your changes missing coverage. Please review.

Project coverage is 87.76%. Comparing base (626aad6) to head (7cc366c).
Report is 2 commits behind head on main.

Files Patch % Lines
src/ott/tools/plot.py 18.84% 56 Missing ⚠️
Additional details and impacted files

Impacted file tree graph

@@            Coverage Diff             @@
##             main     #572      +/-   ##
==========================================
- Coverage   88.38%   87.76%   -0.62%     
==========================================
  Files          72       72              
  Lines        7724     7791      +67     
  Branches     1109     1124      +15     
==========================================
+ Hits         6827     6838      +11     
- Misses        746      802      +56     
  Partials      151      151              
Files Coverage Δ
src/ott/tools/plot.py 54.09% <18.84%> (-21.77%) ⬇️

Copy link
Contributor Author

genious


View entire conversation on ReviewNB

Copy link
Contributor Author

horrible phrasing


View entire conversation on ReviewNB

@michalk8 michalk8 self-requested a review August 21, 2024 12:32
Copy link
Collaborator

@michalk8 michalk8 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, will wait until the CI passes and then merge! Thanks @zoepiran !

@michalk8 michalk8 merged commit aaea0df into ott-jax:main Aug 21, 2024
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet