<filename>rafiki/worker/__init__.py
from .inference import InferenceWorker
from .train import TrainWorker
<filename>rafiki/worker/__init__.py
from .inference import InferenceWorker
from .train import TrainWorker