Make torch import optional for CPU-only Space deployment

server/AxiomForgeAI_environment.py

@@ -46,7 +46,6 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional
 from uuid import uuid4
 
-import torch
 from openenv.core.env_server.interfaces import Environment
 from openenv.core.env_server.types import State
 
@@ -59,12 +58,14 @@ except ImportError:
 # ── Heavy RL imports — fail gracefully so openenv validate passes even when
 #    the ML stack is not installed (e.g. lightweight CI / schema validation).
 try:
+    import torch
     from src.rl.math_environment_curriculum import CurriculumMathEnvironment
     from src.rl.prm_scorer import ProcessRewardScorer
     from src.sft.solution_format import extract_final_answer_numeric_str
 
     _RL_AVAILABLE = True
 except Exception as _rl_import_err:  # pragma: no cover
+    torch = None  # type: ignore[assignment]
     _RL_AVAILABLE = False
     CurriculumMathEnvironment = None  # type: ignore[assignment,misc]
     ProcessRewardScorer = None  # type: ignore[assignment,misc]
@@ -133,7 +134,10 @@ class AxiomforgeaiEnvironment(Environment):
 
         self._math_env: Optional[Any] = None  # CurriculumMathEnvironment or None
 
-        device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        if torch is not None:
+            device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        else:
+            device = "cpu"
 
         if not _RL_AVAILABLE:
             logger.warning(