Upload folder using huggingface_hub

server/app.py

@@ -35,11 +35,8 @@ except Exception as e:  # pragma: no cover
         "openenv is required for the web interface. Install dependencies with '\n    uv sync\n'"
     ) from e
 
-# Support both installed package and direct Docker execution
-try:
-    from benchmark.models import BenchmarkAction, BenchmarkObservation
-except ImportError:
-    from models import BenchmarkAction, BenchmarkObservation
+# Import from local models.py (PYTHONPATH includes /app/env in Docker)
+from models import BenchmarkAction, BenchmarkObservation
 
 from .benchmark_environment import BenchmarkEnvironment
 

server/benchmark_environment.py

@@ -21,11 +21,8 @@ from uuid import uuid4
 from openenv.core.env_server.interfaces import Environment
 from openenv.core.env_server.types import State
 
-# Support both installed package and direct Docker execution
-try:
-    from benchmark.models import BenchmarkAction, BenchmarkObservation
-except ImportError:
-    from models import BenchmarkAction, BenchmarkObservation
+# Import from local models.py (PYTHONPATH includes /app/env in Docker)
+from models import BenchmarkAction, BenchmarkObservation
 
 
 def _get_host_url() -> str: