✅ add TP with CB test

[prashantgupta24]

Description

Adding TP to CB tests

Related Issues

[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 🚀

[joerunde]

couple small comments but as-is this addresses the need for more TP test cases with continuous batching 👍

[joerunde]

TypeError: generate_spyre_vllm_output() missing 1 required positional argument: 'block_size

oof :(

I'd remove from generate_spyre_vllm_output entirely, but also fine w/ passing any old dummy value in