Merge pull request #691 from plasmashen/plasmashen-patch-1

Update attacked_text.py
QData · Sep 23, 2022 · e06f237 · e06f237
2 parents 899ea46 + ef270fc
commit e06f237
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/textattack/shared/attacked_text.py b/textattack/shared/attacked_text.py
@@ -91,8 +91,9 @@ def __eq__(self, other):
                 elif not (self.attack_attrs[key] == other.attack_attrs[key]).all():
                     return False
             else:
-                if not self.attack_attrs[key] == other.attack_attrs[key]:
-                    return False
+                if isinstance(self.attack_attrs[key], AttackedText):
+                    if not self.attack_attrs[key]._text_input == other.attack_attrs[key]._text_input:
+                        return False
         return True
 
     def __hash__(self):