Fix/tests

[michalk8]

No description provided.

[review-notebook-app]

Check out this pull request on 

See visual diffs & provide feedback on Jupyter Notebooks.

---

Powered by ReviewNB

[codecov]

Codecov Report

❌ Patch coverage is 42.85714% with 4 lines in your changes missing coverage. Please review.
✅ Project coverage is 87.23%. Comparing base (2f93834) to head (5a4ac55).

Files with missing lines	Patch %	Lines
src/ott/geometry/geometry.py	0.00%	0 Missing and 1 partial ⚠️
src/ott/geometry/low_rank.py	0.00%	0 Missing and 1 partial ⚠️
src/ott/geometry/pointcloud.py	0.00%	0 Missing and 1 partial ⚠️
src/ott/solvers/linear/lineax_implicit.py	66.66%	0 Missing and 1 partial ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #673      +/-   ##
==========================================
- Coverage   87.32%   87.23%   -0.09%     
==========================================
  Files          82       82              
  Lines        8480     8476       -4     
  Branches      581      581              
==========================================
- Hits         7405     7394      -11     
- Misses        924      931       +7     
  Partials      151      151              

Files with missing lines	Coverage Δ
src/ott/datasets.py	95.00% <ø> (ø)
src/ott/experimental/mmsinkhorn.py	91.56% <ø> (ø)
src/ott/geometry/costs.py	96.54% <ø> (ø)
src/ott/geometry/distrib_costs.py	100.00% <ø> (ø)
src/ott/geometry/epsilon_scheduler.py	95.83% <ø> (ø)
src/ott/geometry/geodesic.py	94.39% <ø> (ø)
src/ott/geometry/graph.py	95.83% <ø> (ø)
src/ott/geometry/grid.py	96.21% <ø> (ø)
src/ott/geometry/regularizers.py	98.65% <ø> (ø)
src/ott/geometry/segment.py	100.00% <ø> (ø)
... and 71 more

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.