__init__.py

from .csvtrace import CSVTrace
from .randomtrace import RandomTrace

types = {
    'csv': CSVTrace,
    'random': RandomTrace
}