physix/training/__init__.py

"""Training utilities for PhysiX-Live.

The ``loop`` submodule pulls in heavy ML deps (torch, unsloth, trl) and is
imported lazily on demand. The lighter prompt + scorer surface is exposed
here so callers without CUDA can still build datasets and score completions.
"""

from physix.training.prompt import (
    build_prompt,
    parse_completion,
    render_observation_for_prompt,
)

__all__ = [
    "build_prompt",
    "parse_completion",
    "render_observation_for_prompt",
]