Skip to content

Commit e52eb3c

Browse files
authored
make sure samples is a list (#609)
1 parent 6be4d6b commit e52eb3c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/setfit/model_card.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -298,7 +298,7 @@ def set_widget_examples(self, dataset: Dataset) -> None:
298298
samples = dataset.select(random.sample(range(len(dataset)), k=min(len(dataset), 5)))["text"]
299299
self.widget = [{"text": sample} for sample in samples]
300300

301-
samples.sort(key=len)
301+
samples = sorted(list(samples), key=len)
302302
if samples:
303303
self.predict_example = samples[0]
304304

0 commit comments

Comments
 (0)