Merge pull request #746 from wenh06/fix_filter_by_labels_

QData · Jul 24, 2023 · 00adb8a · 00adb8a
2 parents 7d7e500 + dabb8a9
commit 00adb8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/textattack/datasets/dataset.py b/textattack/datasets/dataset.py
@@ -125,7 +125,7 @@ def filter_by_labels_(self, labels_to_keep):
         """
         if not isinstance(labels_to_keep, set):
             labels_to_keep = set(labels_to_keep)
-        self._dataset = filter(lambda x: x[1] in labels_to_keep, self._dataset)
+        self._dataset = list(filter(lambda x: x[1] in labels_to_keep, self._dataset))
 
     def __getitem__(self, i):
         """Return i-th sample."""