tests fix

Author: shreymodi1

tests/pytest/test_openenv_browsergym_basic.py

@@ -12,6 +12,12 @@
 # Skip these integration-heavy tests on CI runners by default
 pytestmark = pytest.mark.skipif(os.getenv("CI") == "true", reason="Skip OpenEnv integration tests on CI")
 
+# Skip if OpenEnv not installed
+try:
+    from envs.browsergym_env import BrowserGymEnv, BrowserGymAction  # type: ignore
+except ImportError:
+    pytest.skip("OpenEnv browsergym_env not installed", allow_module_level=True)
+
 
 @pytest.mark.integration
 def test_openenv_browsergym_basic():
@@ -43,8 +49,6 @@ def test_openenv_browsergym_basic():
 
     # Construct the processor with a trivial action_parser; the model output will still be generated
     # but we parse to a safe noop action to minimize flakiness for the environment step.
-    from envs.browsergym_env import BrowserGymAction, BrowserGymEnv  # type: ignore
-
     processor = OpenEnvRolloutProcessor(
         env_factory=None,
         prompt_builder=lambda obs, step, history: "Do nothing",

tests/pytest/test_openenv_browsergym_eval.py

@@ -10,6 +10,12 @@
 # Skip these integration-heavy tests on CI runners by default
 pytestmark = pytest.mark.skipif(os.getenv("CI") == "true", reason="Skip OpenEnv integration tests on CI")
 
+# Skip if OpenEnv not installed
+try:
+    from envs.browsergym_env import BrowserGymEnv, BrowserGymAction  # type: ignore
+except ImportError:
+    pytest.skip("OpenEnv browsergym_env not installed", allow_module_level=True)
+
 
 def openenv_dataset_to_rows(data: List[Dict[str, Any]]) -> List[EvaluationRow]:
     """

tests/pytest/test_openenv_echo_hub.py

@@ -9,13 +9,19 @@
 import pytest
 
 
-# Preferred import when using the monolithic `openenv` package
-from envs.echo_env import EchoEnv  # type: ignore
-
-
 # Skip these integration-heavy tests on CI runners by default
 pytestmark = pytest.mark.skipif(os.getenv("CI") == "true", reason="Skip OpenEnv integration tests on CI")
 
+# Try to import, but skip if not available
+try:
+    from envs.echo_env import EchoEnv  # type: ignore
+
+    _HAS_ECHO = True
+except ImportError:
+    _HAS_ECHO = False
+    EchoEnv = None  # type: ignore
+    pytest.skip("OpenEnv echo_env not installed", allow_module_level=True)
+
 
 def echo_dataset_to_rows(data: List[Dict[str, Any]]) -> List[EvaluationRow]:
     """
@@ -79,7 +85,7 @@ def action_parser(response_text: str):
     rollout_processor=(
         OpenEnvRolloutProcessor(
             # Use HF Hub to launch the environment container automatically
-            env_client_cls=EchoEnv,  # type: ignore
+            env_client_cls=EchoEnv if _HAS_ECHO else None,  # type: ignore
             hub_repo_id=os.getenv("OPENENV_ECHO_REPO", "openenv/echo-env"),
             # Simple prompt+parser above
             prompt_builder=prompt_builder,
@@ -88,6 +94,8 @@ def action_parser(response_text: str):
             timeout_ms=5000,
             num_generations=1,
         )
+        if _HAS_ECHO
+        else None
     ),
 )
 def test_openenv_echo_hub(row: EvaluationRow) -> EvaluationRow: