# Backward-compat shim -- module moved to training/models.py
import importlib as _importlib
import sys as _sys
_sys.modules[__name__] = _importlib.import_module("training.models")