🐛 put decode back in warmup

[joerunde]

Description

Fixes a bug introduced in #283 where the decode pass was moved out of the warmup context.

AKA: @prashantgupta24's editor likes to unindent things, and @joerunde likes to view diffs with whitespace changes ignored, so here we are 😅

[joerunde]

bot:test
MARKERS="cb and spyre"

[github-actions]

👋 Hi! Thank you for contributing to vLLM support on Spyre.
Just a reminder: Make sure that your code passes all the linting checks, otherwise your PR won't be able to be merged. To do so, first install the linting requirements, then run format.sh and commit the changes. This can be done with uv directly:

uv sync --frozen --group lint --active --inexact

Or this can be done with pip:

uv pip compile --group lint > requirements-lint.txt
pip install -r requirements-lint.txt
bash format.sh

Now you are good to go 🚀

[prashantgupta24]

Haha, would be great if it was just an indention problem

[prashantgupta24]

Added to #292