wip - fix test - PoolingParams.normalize deprecated in favor of use_activation

nrghosh · nrghosh · commit 85b904b202ad · 2026-01-21T15:09:47.000-08:00
Signed-off-by: Nikhil Ghosh &lt;nikhil@anyscale.com&gt;
diff --git a/python/ray/llm/tests/batch/gpu/stages/test_vllm_engine_stage.py b/python/ray/llm/tests/batch/gpu/stages/test_vllm_engine_stage.py
@@ -348,7 +348,7 @@ async def test_vllm_wrapper_embed(model_opt_125m):
     [
         ({}, True),
         ({"truncate_prompt_tokens": 3}, False),
-        ({"normalize": True}, False),
+        ({"use_activation": True}, False),
     ],
 )
 async def test_vllm_wrapper_embed_pooling_params(

Original file line number	Diff line number	Diff line change
`@@ -348,7 +348,7 @@ async def test_vllm_wrapper_embed(model_opt_125m):`
`348`	`348`	`[`
`349`	`349`	`({}, True),`
`350`	`350`	`({"truncate_prompt_tokens": 3}, False),`
`351`		`- ({"normalize": True}, False),`
	`351`	`+ ({"use_activation": True}, False),`
`352`	`352`	`],`
`353`	`353`	`)`
`354`	`354`	`async def test_vllm_wrapper_embed_pooling_params(`