[unitest] fix the bug in test_sampler (#3157)

zeroRains · web-flow · commit 17f51f0c9294 · 2025-08-04T01:23:25.000-07:00
diff --git a/test/layers/test_sampler.py b/test/layers/test_sampler.py
@@ -56,6 +56,7 @@ def _create_default_sampling_metadata(
         min_dec_lens=paddle.full(shape=[batch_size, 1], fill_value=min_seq_len, dtype="int64"),
         bad_words_token_ids=paddle.full(shape=[batch_size], fill_value=-1, dtype="int64"),
         eos_token_ids=paddle.full(shape=[batch_size], fill_value=-2, dtype="int64"),
+        min_p=paddle.randn([batch_size]),
     )
     return fake_sampling_metadata
 

Original file line number	Diff line number	Diff line change
`@@ -56,6 +56,7 @@ def _create_default_sampling_metadata(`
`56`	`56`	`min_dec_lens=paddle.full(shape=[batch_size, 1], fill_value=min_seq_len, dtype="int64"),`
`57`	`57`	`bad_words_token_ids=paddle.full(shape=[batch_size], fill_value=-1, dtype="int64"),`
`58`	`58`	`eos_token_ids=paddle.full(shape=[batch_size], fill_value=-2, dtype="int64"),`
	`59`	`+ min_p=paddle.randn([batch_size]),`
`59`	`60`	`)`
`60`	`61`	`return fake_sampling_metadata`
`61`	`62`