Skip to content

Commit 717f86a

Browse files
Update tests/defences/detector/poison/test_activation_defence.py
Co-authored-by: Beat Buesser <[email protected]>
1 parent 706dbaf commit 717f86a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/defences/detector/poison/test_activation_defence.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -102,7 +102,8 @@ def test_wrong_parameters_4(self):
102102

103103
@unittest.expectedFailure
104104
def test_wrong_parameters_5(self):
105-
self.defence.set_params(ex_re_threshold=-1)
105+
with self.assertRaises(ValueError):
106+
self.defence.set_params(ex_re_threshold=-1)
106107

107108
def test_activations(self):
108109
(x_train, _), (_, _), (_, _) = self.mnist

0 commit comments

Comments
 (0)