Fix Space runtime imports

Dockerfile

@@ -25,4 +25,4 @@ ENV PYTHONPATH="/app/env:$PYTHONPATH"
 HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:8000/health || exit 1
 
-CMD ["uvicorn", "server.app:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["uvicorn", "CyberSecurity_OWASP.server.app:app", "--host", "0.0.0.0", "--port", "8000"]

evals.py

@@ -5,7 +5,10 @@ from __future__ import annotations
 import difflib
 from typing import Iterable
 
-from .models import CyberSecurityOWASPAction
+try:
+    from .models import CyberSecurityOWASPAction
+except ImportError:  # pragma: no cover
+    from models import CyberSecurityOWASPAction
 
 
 def random_policy() -> Iterable[CyberSecurityOWASPAction]:

rewards.py

@@ -2,7 +2,10 @@
 
 from __future__ import annotations
 
-from .models import CyberSecurityOWASPAction, CyberSecurityOWASPState
+try:
+    from .models import CyberSecurityOWASPAction, CyberSecurityOWASPState
+except ImportError:  # pragma: no cover
+    from models import CyberSecurityOWASPAction, CyberSecurityOWASPState
 
 
 REWARD_KEYS = (

scenario_compiler.py

@@ -6,9 +6,14 @@ import tempfile
 from pathlib import Path
 from typing import Any
 
-from .fixture_generator import visible_workspace_summary
-from .policy_graph import build_invoice_policy
-from .template_renderer import render_fastapi_basic
+try:
+    from .fixture_generator import visible_workspace_summary
+    from .policy_graph import build_invoice_policy
+    from .template_renderer import render_fastapi_basic
+except ImportError:  # pragma: no cover
+    from fixture_generator import visible_workspace_summary
+    from policy_graph import build_invoice_policy
+    from template_renderer import render_fastapi_basic
 
 
 def compile_scenario(seed: int, split: str = "train", difficulty: int = 0) -> dict[str, Any]:

server/app.py

@@ -16,7 +16,7 @@ except Exception as e:  # pragma: no cover
 try:
     from ..models import CyberSecurityOWASPAction, CyberSecurityOWASPObservation
     from .CyberSecurity_OWASP_environment import CybersecurityOwaspEnvironment
-except ModuleNotFoundError:
+except ImportError:
     from models import CyberSecurityOWASPAction, CyberSecurityOWASPObservation
     from server.CyberSecurity_OWASP_environment import CybersecurityOwaspEnvironment
 

validators.py

@@ -5,7 +5,10 @@ from __future__ import annotations
 from pathlib import Path
 from typing import Any
 
-from .models import CyberSecurityOWASPAction, CyberSecurityOWASPState
+try:
+    from .models import CyberSecurityOWASPAction, CyberSecurityOWASPState
+except ImportError:  # pragma: no cover
+    from models import CyberSecurityOWASPAction, CyberSecurityOWASPState
 
 
 BLOCKED_PATH_MARKERS = (