114514

project/env/__init__.py

@@ -0,0 +1,6 @@
+from .simulator_client import SimulatorClient
+from .local_env import HierarchicalEnv, LowLevelEnv, HRLReward, make_local_env
+from .particle_filter import ChannelParticles
+
+__all__ = ["SimulatorClient", "HierarchicalEnv", "LowLevelEnv", "HRLReward",
+           "make_local_env", "ChannelParticles"]

project/env/local_env.py

@@ -0,0 +1,22 @@
+"""兼容入口：本地简化训练环境。
+
+HierarchicalEnv 是高层宏动作环境；LowLevelEnv 是独立八方向导航环境。
+本文件不复制实现，只转发 mcm26b.hrl 下的真实类。
+"""
+from mcm26b.hrl.hierarchical_env import HRLReward, HierarchicalEnv
+from mcm26b.hrl.low_level_env import LowLevelEnv
+
+__all__ = ["HierarchicalEnv", "LowLevelEnv", "HRLReward", "make_local_env"]
+
+
+def make_local_env(*, problem: int = 3, stage: int = 4, seed: int | None = None,
+                   fidelity: str = "production"):
+    """按题目 Prompt 的四阶段编号创建训练环境。"""
+    from mcm26b.hrl.curriculum import STAGES
+    if not 1 <= int(stage) <= len(STAGES):
+        raise ValueError(f"stage 必须在 1..{len(STAGES)}")
+    env = HierarchicalEnv(problem=problem, stage=STAGES[int(stage) - 1],
+                          fidelity=fidelity)
+    if seed is not None:
+        env.reset(seed=seed)
+    return env

project/env/particle_filter.py

@@ -0,0 +1,4 @@
+"""兼容入口：逐频道网格粒子/幸存集。"""
+from mcm26b.particles import ChannelParticles, angular_gap
+
+__all__ = ["ChannelParticles", "angular_gap"]

project/env/simulator_client.py

@@ -0,0 +1,8 @@
+"""兼容入口：真实类见 mcm26b.protocol.client。"""
+from mcm26b.protocol.client import (ActionRejected, ClearOutcome, DeadlineExceeded,
+                                    ExitInfo, MeasureOutcome, ProtocolError,
+                                    ProtocolViolation, SimulatorClient)
+
+__all__ = ["SimulatorClient", "MeasureOutcome", "ClearOutcome", "ExitInfo",
+           "ProtocolError", "ActionRejected", "ProtocolViolation",
+           "DeadlineExceeded"]