Skip to content

Commit 6e9dc7f

Browse files
remove unused
1 parent 0f7473c commit 6e9dc7f

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

src/fairseq2/recipes/lm/_online_finetune/_rewards.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)