openenv/atari_env

1# Copyright (c) Meta Platforms, Inc. and affiliates.2# All rights reserved.3#4# This source code is licensed under the BSD-style license found in the5# LICENSE file in the root directory of this source tree.6 7"""8Atari Environment for OpenEnv.9 10This module provides OpenEnv integration for Atari 2600 games via the11Arcade Learning Environment (ALE).12 13Example:14    >>> from envs.atari_env import AtariEnv, AtariAction15    >>>16    >>> # Connect to a running server or start via Docker17    >>> env = AtariEnv.from_docker_image("atari-env:latest")18    >>>19    >>> # Reset and interact20    >>> result = env.reset()21    >>> result = env.step(AtariAction(action_id=2))  # UP22    >>> print(result.reward, result.done)23    >>>24    >>> # Cleanup25    >>> env.close()26"""27 28from .client import AtariEnv29from .models import AtariAction, AtariObservation, AtariState30 31__all__ = ["AtariEnv", "AtariAction", "AtariObservation", "AtariState"]32