fix: Fix failing unit test

parthchadha · parthchadha · commit 09c45a82836e · 2025-05-20T07:31:14.000-07:00
Signed-off-by: Parth Chadha &lt;pchadha@nvidia.com&gt;
diff --git a/tests/unit/experience/test_rollouts.py b/tests/unit/experience/test_rollouts.py
@@ -221,6 +221,7 @@ def initial_multi_step_calculator_batch(rollout_tokenizer):
     "stop_token_ids": None,
     "stop_strings": None,
     "vllm_cfg": {
+        "async_engine": False,
         "precision": "bfloat16",
         "tensor_parallel_size": 1,
         "max_model_len": 2048,
