alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

add consistent is_black_box attributes

Browse Source
This commit is contained in:
Jin Yong Yoo
2020-10-06 16:42:21 -04:00
parent 79f3f4f8f0
commit 4893d47e6c
6 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
textattack/shared/attack.py
View File

@@ -67,7 +67,9 @@ class Attack:
self.transformation = transformation
if not self.transformation:
raise NameError("Cannot instantiate attack without transformation")
self.is_black_box = getattr(transformation, "is_black_box", True)
self.is_black_box = (
getattr(transformation, "is_black_box", True) or search_method.is_black_box
)
if not self.search_method.check_transformation_compatibility(
self.transformation
@@ -114,7 +116,7 @@ class Attack:
)
)
self.search_method.filter_transformations = self.filter_transformations
if not search_method.is_blackbox:
if not search_method.is_black_box:
self.search_method.get_model = lambda: self.goal_function.model
def clear_cache(self, recursive=True):