sentinel-env / tasks /easy_task.py
Vizxal's picture
submission ready
b13b69d
Raw History Blame Contribute Delete
899 Bytes
"""
Easy Task: Static targets, low risk, 2 sensors, 3 targets.
Used to test basic sensor allocation functionality.
"""
EASY_TASK_CONFIG = {
"num_sensors": 3,
"num_targets": 3,
"max_steps": 20,
"sensor_failure_prob": 0.0,
"targets": [
{"id": 0, "priority": 3, "position": (2, 2), "dynamic": False, "risk": 0.0},
{"id": 1, "priority": 2, "position": (5, 5), "dynamic": False, "risk": 0.0},
{"id": 2, "priority": 1, "position": (8, 3), "dynamic": False, "risk": 0.0},
],
"sensors": [
{"id": 0, "range": 10, "available": True},
{"id": 1, "range": 10, "available": True},
{"id": 2, "range": 10, "available": True},
],
}
def get_easy_task():
return EASY_TASK_CONFIG
def get_easy_env():
from env import SentinelEnv
return SentinelEnv(max_steps=EASY_TASK_CONFIG["max_steps"], seed=42, config=EASY_TASK_CONFIG)