one round of improvements on tests

[AlexanderFengler]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 94.11765% with 2 lines in your changes missing coverage. Please review.

Files with missing lines	Patch %	Lines
src/lanfactory/trainers/torch_mlp.py	90.90%	1 Missing and 1 partial ⚠️

Files with missing lines	Coverage Δ
src/lanfactory/cli/utils.py	100.00% <ø> (ø)
src/lanfactory/onnx/transform_onnx.py	100.00% <ø> (+5.26%)	⬆️
src/lanfactory/trainers/jax_mlp.py	94.68% <100.00%> (+7.18%)	⬆️
src/lanfactory/utils/util_funs.py	100.00% <100.00%> (ø)
src/lanfactory/trainers/torch_mlp.py	86.64% <90.90%> (+7.86%)	⬆️

... and 5 files with indirect coverage changes

🚀 New features to boost your workflow:

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