code/__init__.py

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

"""CropRL Environment — single-agent and multi-agent farm management."""

from .client import CroprlEnv
from .config import EnvConfig, MultiAgentConfig
from .enums import (
    ActionType,
    CropType,
    ForumMsgType,
    HypePhase,
    LedgerEventType,
    Season,
)
from .models import (
    CroprlAction,
    CroprlObservation,
    ForumMessage,
    HypeCropStatus,
    LedgerEvent,
    MultiAgentAction,
    MultiAgentObservation,
    MultiAgentResult,
    MultiAgentState,
)
from .farm_state import FarmState
from .market_engine import MarketEngine, HypeEngine
from .public_ledger import Forum, PublicLedger
from .time_controller import TimeController, TurnOverError
from .multi_agent_environment import MultiAgentCroprlEnvironment
from .tasks import (
    TASKS,
    MultiAgentGrader,
    create_env_for_task,
    create_multi_agent_env_for_task,
    grader,
    list_tasks,
    run_multi_agent_episode,
)

__all__ = [
    # Config
    "EnvConfig",
    "MultiAgentConfig",
    # Enums
    "ActionType",
    "CropType",
    "ForumMsgType",
    "HypePhase",
    "LedgerEventType",
    "Season",
    # Models
    "CroprlAction",
    "CroprlObservation",
    "ForumMessage",
    "HypeCropStatus",
    "LedgerEvent",
    "MultiAgentAction",
    "MultiAgentObservation",
    "MultiAgentResult",
    "MultiAgentState",
    # FarmState
    "FarmState",
    # Engines
    "Forum",
    "HypeEngine",
    "MarketEngine",
    "PublicLedger",
    "TimeController",
    "TurnOverError",
    # Environments
    "CroprlEnv",
    "MultiAgentCroprlEnvironment",
    # Tasks & Grading
    "TASKS",
    "MultiAgentGrader",
    "create_env_for_task",
    "create_multi_agent_env_for_task",
    "grader",
    "list_tasks",
    "run_multi_agent_episode",
]