diff --git a/src/setfit/model_card.py b/src/setfit/model_card.py index a143d316..30706bcc 100644 --- a/src/setfit/model_card.py +++ b/src/setfit/model_card.py @@ -298,7 +298,7 @@ def set_widget_examples(self, dataset: Dataset) -> None: samples = dataset.select(random.sample(range(len(dataset)), k=min(len(dataset), 5)))["text"] self.widget = [{"text": sample} for sample in samples] - samples.sort(key=len) + samples = sorted(list(samples), key=len) if samples: self.predict_example = samples[0]