app/__init__.py

# from app.env import GovWorkflowEnv
from app.models import ActionModel, ObservationModel, RewardModel

try:
    from client import GovWorkflowClient
except ModuleNotFoundError:
    GovWorkflowClient = None  # type: ignore[assignment]

GovWorkflowAction = ActionModel
GovWorkflowObservation = ObservationModel

__all__ = [
    "ActionModel",
    "ObservationModel",
    "RewardModel",
    "GovWorkflowAction",
    "GovWorkflowObservation",
#     "GovWorkflowEnv",
    "GovWorkflowClient",
]