@@ -123,7 +123,6 @@ def process_rollouts(
123123 self ,
124124 vllm_outputs : list [RequestOutput ],
125125 prompt_batch : PromptBatch ,
126- sampling_params : SamplingParams = None
127126 ):
128127 batch_text = []
129128 batch_tokens = []
@@ -248,7 +247,6 @@ def process_rollouts(
248247 self ,
249248 vllm_outputs : list [RequestOutput ],
250249 prompt_batch : PromptBatch ,
251- sampling_params : SamplingParams = None
252250 ):
253251 batch_text = []
254252 batch_tokens = []
@@ -369,7 +367,7 @@ def format_prompt(self, prompt_text, rollout_text):
369367
370368 @override
371369 def process_rollouts (
372- self , vllm_outputs : list [RequestOutput ], prompt_batch : PromptBatch , sampling_params : SamplingParams = None
370+ self , vllm_outputs : list [RequestOutput ], prompt_batch : PromptBatch
373371 ):
374372 vllm_inputs = []
375373 batch_text = []
@@ -550,7 +548,7 @@ def __init__(
550548
551549 @override
552550 def process_rollouts (
553- self , vllm_outputs : list [RequestOutput ], prompt_batch : PromptBatch , sampling_params : SamplingParams = None
551+ self , vllm_outputs : list [RequestOutput ], prompt_batch : PromptBatch
554552 ):
555553 vllm_inputs = []
556554 prefixes = []
@@ -753,7 +751,7 @@ def __init__(
753751
754752 @override
755753 def process_rollouts (
756- self , vllm_outputs : list [RequestOutput ], prompt_batch : PromptBatch , sampling_params : SamplingParams = None
754+ self , vllm_outputs : list [RequestOutput ], prompt_batch : PromptBatch
757755 ):
758756 vllm_inputs = []
759757 batch_text = []
0 commit comments