__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.

"""BrowserForge package."""

try:
    from .client import BrowserEnv
    from .models import (
        BrowserAction,
        BrowserElement,
        BrowserObservation,
        ConstraintState,
        RewardBreakdown,
    )
except ImportError:  # pragma: no cover - direct repo execution / pytest collection
    from client import BrowserEnv
    from models import (
        BrowserAction,
        BrowserElement,
        BrowserObservation,
        ConstraintState,
        RewardBreakdown,
    )

__all__ = [
    "BrowserAction",
    "BrowserElement",
    "BrowserObservation",
    "BrowserEnv",
    "ConstraintState",
    "RewardBreakdown",
]