The HackAtari Environments
HackAtari is an object-centric extension of the Atari Learning Environment (ALE), built on top of OCAtari. It enables custom environment modifications and dynamic reward functions for Atari games in research and experimentation.
Example
Create an HackAtari env and play random moves1# Create an HackAtari environment with ram-based object detection and DQN-like observation 2env = HackAtari(env_name="ALE/Pong-v5", mode="ram", obs_mode="dqn") 3 4# Interact with the environment 5obs = env.reset() 6done = False 7while not done: 8 action = env.action_space.sample() # Sample a random action 9 obs, reward, terminated, truncated, info = env.step(action) 10 done = terminated or truncated 11 12 # Render the environment with object overlays 13 env.render()