# SPDX-License-Identifier: Apache-2.0

from __future__ import annotations

from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from fastvideo.train.methods.fine_tuning.dfsft import DiffusionForcingSFTMethod
    from fastvideo.train.methods.fine_tuning.finetune import FineTuneMethod
    from fastvideo.train.methods.fine_tuning.tfsft import TeacherForcingSFTMethod

__all__ = [
    "DiffusionForcingSFTMethod",
    "FineTuneMethod",
    "TeacherForcingSFTMethod",
]


def __getattr__(name: str) -> object:
    # Lazy import to avoid circular imports during registry bring-up.
    if name == "DiffusionForcingSFTMethod":
        from fastvideo.train.methods.fine_tuning.dfsft import (
            DiffusionForcingSFTMethod, )

        return DiffusionForcingSFTMethod
    if name == "FineTuneMethod":
        from fastvideo.train.methods.fine_tuning.finetune import FineTuneMethod

        return FineTuneMethod
    if name == "TeacherForcingSFTMethod":
        from fastvideo.train.methods.fine_tuning.tfsft import (
            TeacherForcingSFTMethod, )

        return TeacherForcingSFTMethod
    raise AttributeError(name)