Fix imports for flat deployment structure in HF Spaces

inference.py

@@ -271,8 +271,8 @@ async def run_direct_baseline() -> Dict:
     Run baseline directly using environment and grader imports.
     Used by the /baseline endpoint to avoid self-HTTP calls on HF Spaces.
     """
-    from ..server.environment import PolicyEvolverEnvironment
-    from ..server.grader import grade
+    from server.environment import PolicyEvolverEnvironment
+    from server.grader import grade
 
     env = PolicyEvolverEnvironment()
     use_llm = verify_environment()

openenv.yaml

@@ -10,7 +10,7 @@ tags:
   - "AI-safety"
 
 environment:
-  module: "policy_evolver_env.server.environment"
+  module: "server.environment"
   class: "PolicyEvolverEnvironment"
 
   observation_schema:

server/Dockerfile

@@ -17,4 +17,4 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=10s --retries=3 \
   CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:8000/health')"
 
 # Start server
-CMD ["uvicorn", "policy_evolver_env.server.app:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "server.app:app", "--host", "0.0.0.0", "--port", "8000"]

server/app.py

@@ -3,13 +3,13 @@ from __future__ import annotations
 from fastapi import FastAPI, HTTPException, Query
 from fastapi.responses import JSONResponse
 from openenv.core.env_server import create_fastapi_app
-from ..models import (
+from models import (
     ProposeClarificationAction, ProposeNewRuleAction, EvolveProcessAction,
     Observation, Action
 )
-from .environment import PolicyEvolverEnvironment
-from .grader import grade
-from .tasks import TASK_REGISTRY
+from server.environment import PolicyEvolverEnvironment
+from server.grader import grade
+from server.tasks import TASK_REGISTRY
 import json
 
 # Create app via OpenEnv helper — pass factory callable, action/obs classes
@@ -58,6 +58,6 @@ async def run_baseline_endpoint():
     Runs the LLM baseline (or rule-based fallback) and returns scores for all tasks.
     Uses the grader directly instead of HTTP calls to self.
     """
-    from ..inference import run_direct_baseline
+    from inference import run_direct_baseline
     results = await run_direct_baseline()
     return results

server/environment.py

@@ -4,12 +4,12 @@ import uuid
 import random
 from typing import Optional, Any, Dict
 from openenv.core.env_server import Environment
-from ..models import (
+from models import (
     Action, Observation, State,
     ProposeClarificationAction, ProposeNewRuleAction, EvolveProcessAction,
 )
-from .grader import grade
-from .tasks import TASK_REGISTRY
+from server.grader import grade
+from server.tasks import TASK_REGISTRY
 
 
 class PolicyEvolverEnvironment(Environment[Action, Observation, State]):

server/grader.py

@@ -6,11 +6,11 @@ All functions return float in [0.0, 1.0].
 from __future__ import annotations
 import re
 from typing import Dict, List, Any
-from ..models import (
+from models import (
     ProposeClarificationAction, ProposeNewRuleAction, EvolveProcessAction,
     Observation
 )
-from .tasks import TASK_REGISTRY
+from server.tasks import TASK_REGISTRY
 
 
 # ─────────────────────────────────────────────