_tmp_notebook_patch_check/training/__init__.py

from .config import TrainingConfig, DEFAULT_CONFIG
from .train import train, run_episode, run_random_baseline