passing batch_size fix

[ArinaDanilina]

No description provided.

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Comparison is base (05530db) 78.52% compared to head (cd0e301) 78.55%.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #647      +/-   ##
==========================================
+ Coverage   78.52%   78.55%   +0.03%     
==========================================
  Files          36       36              
  Lines        3828     3829       +1     
  Branches      672      672              
==========================================
+ Hits         3006     3008       +2     
  Misses        547      547              
+ Partials      275      274       -1     

Files	Coverage Δ
src/moscot/base/problems/_mixins.py	85.37% <100.00%> (+0.54%)	⬆️
src/moscot/problems/cross_modality/_mixins.py	83.60% <ø> (ø)
src/moscot/problems/space/_mixins.py	81.64% <ø> (ø)
src/moscot/problems/time/_mixins.py	63.26% <ø> (ø)

[MUCDK]

minor changes, then we can merge. thanks

[MUCDK]

You write everywhere

Number of rows/columns of the cost matrix to 
materialize during :meth:`push` or :meth:`pull`.
If :obj:`None`, the entire cost matrix will be materialized.
Larger value will require more memory.

Please swap the 3rd with the 4th line. First talk about non-None, and then about None.

[MUCDK]

Thanks!