Fix bug

sanising · sanising · commit f9355d4bde99 · 2025-11-20T17:48:46.000-06:00
Signed-off-by: sanising &lt;sanising@qti.qualcomm.com&gt;
diff --git a/tests/transformers/sampler/test_sampler.py b/tests/transformers/sampler/test_sampler.py
@@ -639,6 +639,7 @@ def test_guided_decoding(
                     (full_batch_size, 1),
                 )
             },
+            **additional_params,
         },
     )
     model_w_sampler_wo_guided_decoding_exec_info = model_w_sampler_wo_guided_decoding.generate(
@@ -648,6 +649,7 @@ def test_guided_decoding(
         include_sampler=True,
         return_pdfs=False,
         sampling_params=sampling_params,
+        **additional_params,
     )
     assert (
         model_w_sampler_w_guided_decoding_exec_info.generated_ids

Original file line number	Diff line number	Diff line change
`@@ -639,6 +639,7 @@ def test_guided_decoding(`
`639`	`639`	`(full_batch_size, 1),`
`640`	`640`	`)`
`641`	`641`	`},`
	`642`	`+ **additional_params,`
`642`	`643`	`},`
`643`	`644`	`)`
`644`	`645`	`model_w_sampler_wo_guided_decoding_exec_info = model_w_sampler_wo_guided_decoding.generate(`
`@@ -648,6 +649,7 @@ def test_guided_decoding(`
`648`	`649`	`include_sampler=True,`
`649`	`650`	`return_pdfs=False,`
`650`	`651`	`sampling_params=sampling_params,`
	`652`	`+ **additional_params,`
`651`	`653`	`)`
`652`	`654`	`assert (`
`653`	`655`	`model_w_sampler_w_guided_decoding_exec_info.generated_ids`