__init__.py

from .bubble import bubble
from .direct import direct
from .epidemic import epidemic
from .flooding import flooding
from .hcbf import hcbf

types = {
    'bubble': bubble,
    'direct': direct,
    'epidemic': epidemic,
    'flooding': flooding,
    'hcbf': hcbf,
}