Skip to content

Commit f6dce36

Browse files
author
Beat Buesser
committed
Update tests
Signed-off-by: Beat Buesser <[email protected]>
1 parent d846fed commit f6dce36

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

art/attacks/attack.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def poison(self, x, y=None, **kwargs):
167167
raise NotImplementedError
168168

169169

170-
class PoisoningAttackWhiteBox(PoisoningAttackBlackBox):
170+
class PoisoningAttackWhiteBox(Attack):
171171
"""
172172
Abstract base class for poisoning attack classes that have white-box access to the model (classifier object)
173173
"""

tests/attacks/test_backdoor_attack.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ class TestBackdoorAttack(TestBase):
4242

4343
def setUp(self):
4444
master_seed(seed=301)
45-
self.backdoor_path = os.path.join("data", "backdoors", "alert.png")
45+
self.backdoor_path = os.path.join(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__)))),
46+
"data", "backdoors", "alert.png")
4647
super().setUp()
4748

4849
@staticmethod

0 commit comments

Comments
 (0)