(#7) Attack: Typo fix

betarixm · Apr 18, 2022 · f52f374 · f52f374
1 parent 8eca0d7
commit f52f374
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/attack/attack.py b/src/attack/attack.py
@@ -49,9 +49,7 @@
     attack_cls: Attack.__class__
 
     if args.dataset == "mnist":
-        classifier = (
-            Classifier(name="victim_classifier_mnist", input_shape=(28, 28, 1)),
-        )
+        classifier = Classifier(name="victim_classifier_mnist", input_shape=(28, 28, 1))
 
         dataset = Mnist()