Skip to content

Commit c8be278

Browse files
committed
fix rebase errors 16
1 parent fc35bbe commit c8be278

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

inference_perf/datagen/shared_prefix_datagen.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ def _generate_prompts(self) -> None:
176176
random.shuffle(self.user_sessions)
177177
else:
178178
# Shuffle prompts and output lengths in sync
179-
combined = list(zip(self.prompts, self.flat_output_lens))
179+
combined = list(zip(self.prompts, self.flat_output_lens, strict=True))
180180
random.shuffle(combined)
181181
self.prompts, self.flat_output_lens = [list(t) for t in zip(*combined)]
182182

0 commit comments

Comments
 (0)