Merge pull request #140 from meta-pytorch/add-browsergym-env-ben

[ENHANCEMENT] Fix browsergym environment

Author: init27

src/envs/browsergym_env/server/Dockerfile

@@ -22,6 +22,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libcairo2 \
     libasound2 \
     libxshmfence1 \
+    fonts-unifont \
+    fonts-noto-color-emoji \
     # Additional dependencies
     git \
     wget \
@@ -33,8 +35,7 @@ RUN pip install --no-cache-dir -r /tmp/browsergym_requirements.txt && \
     rm /tmp/browsergym_requirements.txt
 
 # Install Playwright browsers (Chromium by default)
-RUN playwright install chromium && \
-    playwright install-deps chromium
+RUN playwright install chromium
 
 # Copy OpenEnv core and browsergym_env code
 WORKDIR /app

src/envs/browsergym_env/server/browsergym_environment.py

@@ -8,6 +8,7 @@
 - WorkArena: Enterprise task automation
 """
 
+import importlib
 from typing import Any, Dict, Optional
 from uuid import uuid4
 
@@ -66,6 +67,25 @@ def __init__(
         else:
             self.env_id = f"browsergym/{benchmark}"
 
+        # force import the benchmark module
+        benchmark_modules = {
+            "miniwob": "browsergym.envs.miniwob",
+            "webarena": "browsergym.envs.webarena",
+            "visualwebarena": "browsergym.envs.visualwebarena",
+            "workarena": "browsergym.envs.workarena",
+        }
+        module_path = benchmark_modules.get(benchmark)
+        try:
+            if module_path:
+                importlib.import_module(module_path)
+            else:
+                importlib.import_module("browsergym")
+        except ModuleNotFoundError as import_error:
+            raise ValueError(
+                f"Failed to import BrowserGym benchmark '{benchmark}': {import_error}\n"
+                f"Make sure the package browsergym-{benchmark} is installed."
+            ) from import_error
+
         # Create the BrowserGym environment
         try:
             self.gym_env = gym.make(