Rollout extrapolation steps fix

[gerkone]

No description provided.

[codecov]

Codecov Report

Attention: 4 lines in your changes are missing coverage. Please review.

Comparison is base (d0e5133) 69.5% compared to head (b03cd06) 69.4%.

Files	Patch %	Lines
lagrangebench/evaluate/rollout.py	20.0%	4 Missing ⚠️

Additional details and impacted files

@@           Coverage Diff           @@
##            main     #25     +/-   ##
=======================================
- Coverage   69.5%   69.4%   -0.1%     
=======================================
  Files         25      25             
  Lines       1660    1661      +1     
=======================================
  Hits        1154    1154             
- Misses       506     507      +1     

Flag	Coverage Δ
unittests	69.4% <20.0%> (-0.1%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[arturtoshev]

Looks good!