Upload New File

src/attacks/__init__.py

+# This init file makes it easy to import the main attack classes
+# all at once from the attacks package.
+
+from .base_attack import BaseAttack
+from .fgsm import FGSMAttack
+from .pgd import PGDAttack
+
+# By listing them here, other scripts can do:
+# from attacks import BaseAttack, FGSMAttack, PGDAttack
+__all__ = ["BaseAttack", "FGSMAttack", "PGDAttack"]