inference/__init__.py

from inference.metrics import EpisodeMetrics
from inference.model_wrapper import ModelWrapper

__all__ = ["EpisodeMetrics", "ModelWrapper"]