__init__.py

import importlib
import os


# ASG loss requires wav2letter
blacklist = set()
try:
    import wav2letter
except ImportError:
    blacklist.add("ASG_loss.py")

for file in os.listdir(os.path.dirname(__file__)):
    if file.endswith(".py") and not file.startswith("_") and file not in blacklist:
        criterion_name = file[: file.find(".py")]
        importlib.import_module(
            "examples.speech_recognition.criterions." + criterion_name
        )