Merge branch 'meta-pytorch:main' into main

Author: burtenshaw

.github/workflows/deploy-hf-env.yml

@@ -15,6 +15,11 @@ on:
           - 'chat_env'
           - 'atari_env'
           - 'openspiel_env'
+      custom_environment:
+        description: 'Custom environment to deploy (leave empty for none)'
+        required: false
+        type: string
+        default: ''
       base_image_sha:
         description: 'SHA for openenv-base image (leave empty for latest)'
         required: false
@@ -36,9 +41,6 @@ on:
         type: string
         default: ''
 
-env:
-  HF_USERNAME: openenv
-
 jobs:
   # Job to determine which environments to deploy
   determine-environments:
@@ -124,7 +126,8 @@ jobs:
       - name: Deploy to Hugging Face Space
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
-          HF_NAMESPACE: ${{ env.HF_USERNAME }}
+          HF_NAMESPACE: ${{ github.event.inputs.hf_namespace }}
+          HF_USERNAME: ${{ github.event.inputs.hf_namespace }}
         run: |
           chmod +x scripts/deploy_to_hf.sh
           if [ -n "${{ github.event.inputs.base_image_sha }}" ]; then
@@ -156,7 +159,8 @@ jobs:
       - name: Deploy to Hugging Face Space
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
-          HF_NAMESPACE: ${{ env.HF_USERNAME }}
+          HF_NAMESPACE: ${{ github.event.inputs.hf_namespace }}
+          HF_USERNAME: ${{ github.event.inputs.hf_namespace }}
         run: |
           chmod +x scripts/deploy_to_hf.sh
           if [ -n "${{ github.event.inputs.base_image_sha }}" ]; then

.github/workflows/docker-build.yml

@@ -79,6 +79,8 @@ jobs:
             dockerfile: src/envs/atari_env/server/Dockerfile
           - name: git-env
             dockerfile: src/envs/git_env/server/Dockerfile
+          - name: textarena-env
+            dockerfile: src/envs/textarena_env/server/Dockerfile
 
     steps:
       - name: Checkout code

.github/workflows/pr-new-env-health-check.yml

@@ -0,0 +1 @@
+name: PR New Environment

examples/atari_simple.py

@@ -16,7 +16,14 @@
     python examples/atari_simple.py
 """
 
+import sys
+from pathlib import Path
+
 import numpy as np
+
+# Add src to path
+sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
+
 from envs.atari_env import AtariEnv, AtariAction