Skip to content

Commit 8ce6a13

Browse files
committed
[ci] use cat as default input
1 parent c1e3dfa commit 8ce6a13

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_simba.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,15 @@ class TestSimBA(unittest.TestCase):
1313

1414
def test_simba(self):
1515
# Load Image [0.0, 1.0]
16-
x = np.asarray(Image.open("tests/dog.jpg").resize((32, 32))) / 255.0
16+
x = np.asarray(Image.open("tests/cat.jpg").resize((32, 32))) / 255.0
1717

1818
# Initialize API Model
1919
model = VGG16Cifar10("https://api.wuhanstudio.uk" + "/vgg16_cifar10")
2020

2121
# Get Preditction
2222
y_pred = model.predict(np.array([x]))[0]
2323

24-
assert (np.argmax(y_pred) == 5)
24+
assert (np.argmax(y_pred) == 3)
2525

2626
# SimBA Attack
2727
simba = SimBA(model)

0 commit comments

Comments
 (0)