app/env/__init__.py

"""Environment package."""

__all__ = ["PolyGuardEnv", "EnvironmentA", "EnvironmentB", "EnvironmentC", "EnvironmentD"]


def __getattr__(name: str):
    if name == "PolyGuardEnv":
        from app.env.env_core import PolyGuardEnv

        return PolyGuardEnv
    if name == "EnvironmentA":
        from app.env.environment_a import EnvironmentA

        return EnvironmentA
    if name == "EnvironmentB":
        from app.env.environment_b import EnvironmentB

        return EnvironmentB
    if name == "EnvironmentC":
        from app.env.environment_c import EnvironmentC

        return EnvironmentC
    if name == "EnvironmentD":
        from app.env.environment_d import EnvironmentD

        return EnvironmentD
    raise AttributeError(name)