__init__.py

try:
    from .client import main
    from .models import Action
    from .models import EnvironmentState
    from .models import Observation
except ImportError:
    from client import main
    from models import Action
    from models import EnvironmentState
    from models import Observation

__all__ = [
    "Action",
    "Observation",
    "EnvironmentState",
    "main",
]