We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 41c779b + 60a9e50 commit 1be82ddCopy full SHA for 1be82dd
benchmark/benchmark.py
@@ -28,7 +28,8 @@ def sample_texts(
28
dataset = [data for data in dataset if len(data["conversations"]) >= 2]
29
# Keep the first two turns of each conversation
30
dataset = [
31
- (data["conversations"][0]["value"], data["conversations"][1]["value"]) for data in sample(dataset, k=num_texts)
+ (data["conversations"][0]["value"], data["conversations"][1]["value"])
32
+ for data in sample(dataset, k=num_texts) # nosec
33
]
34
shuffle(dataset)
35
return dataset
0 commit comments