Misc/update typing

[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 98.63714% with 8 lines in your changes missing coverage. Please review.
✅ Project coverage is 86.88%. Comparing base (98612bd) to head (cc6da40).

Files with missing lines	Patch %	Lines
src/ott/neural/methods/expectile_neural_dual.py	75.00%	2 Missing ⚠️
src/ott/neural/networks/potentials.py	84.61%	2 Missing ⚠️
src/ott/data/_loaders.py	0.00%	1 Missing ⚠️
src/ott/experimental/mmsinkhorn.py	95.00%	0 Missing and 1 partial ⚠️
src/ott/math/_velocity_from_brenier_potential.py	66.66%	1 Missing ⚠️
src/ott/tools/progot.py	83.33%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #708      +/-   ##
==========================================
- Coverage   87.00%   86.88%   -0.13%     
==========================================
  Files          85       85              
  Lines        8888     8753     -135     
  Branches      616      614       -2     
==========================================
- Hits         7733     7605     -128     
+ Misses        996      990       -6     
+ Partials      159      158       -1     

Files with missing lines	Coverage Δ
src/ott/datasets.py	97.61% <100.00%> (-0.32%)	⬇️
src/ott/geometry/costs.py	96.55% <100.00%> (+<0.01%)	⬆️
src/ott/geometry/distrib_costs.py	100.00% <100.00%> (ø)
src/ott/geometry/epsilon_scheduler.py	95.65% <100.00%> (-0.19%)	⬇️
src/ott/geometry/geodesic.py	94.49% <100.00%> (+0.10%)	⬆️
src/ott/geometry/geometry.py	90.45% <100.00%> (+0.03%)	⬆️
src/ott/geometry/graph.py	95.87% <100.00%> (+0.04%)	⬆️
src/ott/geometry/grid.py	96.24% <100.00%> (+0.02%)	⬆️
src/ott/geometry/low_rank.py	96.42% <100.00%> (+0.02%)	⬆️
src/ott/geometry/pointcloud.py	94.09% <100.00%> (+0.02%)	⬆️
... and 74 more

🚀 New features to boost your workflow:

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