Add predict kwargs in validation step

[LeonStadelmann]

No description provided.

[LeonStadelmann]

If tests are necessary, should they be added in test_cellflow or in test_trainer?

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 81.82%. Comparing base (4970ad4) to head (c281d9a).
Report is 11 commits behind head on main.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #228      +/-   ##
==========================================
+ Coverage   81.35%   81.82%   +0.47%     
==========================================
  Files          38       38              
  Lines        2564     2548      -16     
  Branches      329      321       -8     
==========================================
- Hits         2086     2085       -1     
+ Misses        334      328       -6     
+ Partials      144      135       -9     

Files with missing lines	Coverage Δ
src/cellflow/model/_cellflow.py	73.95% <100.00%> (+1.93%)	⬆️
src/cellflow/training/_trainer.py	98.30% <100.00%> (+0.05%)	⬆️

... and 2 files with indirect coverage changes

🚀 New features to boost your workflow:

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

[MUCDK]

If tests are necessary, should they be added in test_cellflow or in test_trainer?

Let's please add a test checking for passing the number of steps of the solver, and make sure if it's 2 steps it's considerablyu faster (e.g. 1 sec) than passing 1e7 steps.

Let's add it to test_trainer

[MUCDK]

Looks great, just add tests!

[MUCDK]

Great, thanks!

[MUCDK]

@LeonStadelmann thanks a lot, looks great!
Please resolve conflicts with main, then we can merge!