[Bugfix] Fix task field initialization when PYTHONOPTIMIZE is enabled (vllm-project#23718)

cndoit18 · amd-xiaoyu12 · commit ba5ccd18a9f0 · 2025-08-28T15:54:27.000Z
Signed-off-by: cndoit18 &lt;cndoit18@outlook.com&gt;
Signed-off-by: Xiao Yu &lt;xiao.yu@amd.com&gt;
diff --git a/vllm/worker/pooling_model_runner.py b/vllm/worker/pooling_model_runner.py
@@ -199,8 +199,9 @@ def _prepare_pooling(
 
             pooling_params = seq_group_metadata.pooling_params
             assert pooling_params is not None
-            assert (task := pooling_params.task) is not None, (
-                "You did not set `task` in the API")
+
+            task = pooling_params.task
+            assert task is not None, "You did not set `task` in the API"
 
             model = cast(VllmModelForPooling, self.model)
             to_update = model.pooler.get_pooling_updates(task)
