Skip to content

Commit 1be82dd

Browse files
authored
Merge branch 'master' into dependabot/github_actions/actions/cache-5.0.0
2 parents 41c779b + 60a9e50 commit 1be82dd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

benchmark/benchmark.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ def sample_texts(
2828
dataset = [data for data in dataset if len(data["conversations"]) >= 2]
2929
# Keep the first two turns of each conversation
3030
dataset = [
31-
(data["conversations"][0]["value"], data["conversations"][1]["value"]) for data in sample(dataset, k=num_texts)
31+
(data["conversations"][0]["value"], data["conversations"][1]["value"])
32+
for data in sample(dataset, k=num_texts) # nosec
3233
]
3334
shuffle(dataset)
3435
return dataset

0 commit comments

Comments
 (0)