fix(grpc): derive choice count before normalization (#35778)

Signed-off-by: Connor Carpenter <connorc@nvidia.com>
2026-08-21 11:29:52 -07:00
parent 61c2da42bb
commit c3735625de
2 changed files with 4 additions and 2 deletions
@@ -93,7 +93,7 @@ class TestNativeGrpcParallelResponses(CustomTestCase):
            },
        ]
        handle = _make_runtime_handle(responses)
-        obj = SimpleNamespace(rid="logical", batch_size=1, parallel_sample_num=2)
+        obj = SimpleNamespace(rid="logical", sampling_params={"n": 2})

        asyncio.run(
            handle._run_generate(