Merge pull request #75 from burtenshaw/fix/atari-example

Fix atari example script

Author: init27

examples/atari_simple.py

@@ -24,7 +24,7 @@ def main():
     """Run a simple Atari episode."""
     # Connect to the Atari environment server
     print("Connecting to Atari environment...")
-    env = AtariEnv(base_url="http://localhost:8000")
+    env = AtariEnv.from_docker_image("ghcr.io/meta-pytorch/openenv-atari-env:latest")
 
     try:
         # Reset the environment
@@ -42,7 +42,8 @@ def main():
         for step in range(100):
             # Random action
             action_id = np.random.choice(result.observation.legal_actions)
-
+            action_id = int(action_id)
+            
             # Take action
             result = env.step(AtariAction(action_id=action_id))