src/training/__init__.py

"""KAPS Training Module"""
from .kaps_env import KAPSEnv
from .train_dreamer import train_dreamer

__all__ = ['KAPSEnv', 'train_dreamer']