Skip to content

Commit 7f4a993

Browse files
authored
Merge pull request #813 from mys007/fix-composite-transform-determinism
Fix composite transform reproducibility
2 parents 96b92be + fc98df7 commit 7f4a993

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

textattack/transformations/composite_transformation.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -34,10 +34,11 @@ def _get_transformations(self, *_):
3434
)
3535

3636
def __call__(self, *args, **kwargs):
37-
new_attacked_texts = set()
37+
new_attacked_texts = []
3838
for transformation in self.transformations:
39-
new_attacked_texts.update(transformation(*args, **kwargs))
40-
return list(new_attacked_texts)
39+
new_attacked_texts.extend(transformation(*args, **kwargs))
40+
seen = set()
41+
return [t for t in new_attacked_texts if not (t in seen or seen.add(t))]
4142

4243
def __repr__(self):
4344
main_str = "CompositeTransformation" + "("

0 commit comments

Comments
 (0)