ID2T-toolkit/code/Attack/__init__.py

import glob
import os.path as osp

# Create a list consisting of all attack classes names in this package
__all__ = []
modules = glob.glob(osp.dirname(__file__) + "/*.py")
for m in modules:
    c = str(osp.basename(m)[:-3])
    if not c.startswith('__') and not c.startswith('Base') and c.endswith('Attack'):
        __all__.append(c)