Skip to content

Commit 60a9e50

Browse files
authored
Avoid Bandit False Positive (#591)
1 parent a4d8fdf commit 60a9e50

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)