ctm-dqn/test_optimized_route.py

11 lines
290 B
Python

from sumo_vsl_environment import SUMOVSLEnvironment
import yaml
with open('config_sumo_vsl.yaml', 'r', encoding='utf-8') as f:
config = yaml.safe_load(f)
env = SUMOVSLEnvironment(config)
state = env.reset(seed=42)
print(f"测试成功! 初始状态维度: {state.shape}")
env.close()