Skip to content

Commit b0c00b4

Browse files
Update tests/attacks/poison/test_one_pixel_shortcut_attack.py
Co-authored-by: Beat Buesser <[email protected]>
1 parent 4d074f5 commit b0c00b4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/attacks/poison/test_one_pixel_shortcut_attack.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def test_one_pixel_effect_with_pytorchclassifier():
139139
acc_poisoned = np.mean(preds_poisoned.argmax(axis=1) == y_poison)
140140

141141
# Adjusted assertions for robustness
142-
assert acc_poisoned >= 1.0, f"Expected 100% poisoned accuracy, got {acc_poisoned:.3f}"
142+
assert acc_poisoned == 1.0, f"Expected 100% poisoned accuracy, got {acc_poisoned:.3f}"
143143
assert acc_clean < 0.95, f"Expected clean accuracy < 95%, got {acc_clean:.3f}"
144144

145145
except Exception as e:

0 commit comments

Comments
 (0)