Spaces:
Sleeping
Sleeping
Commit ·
aea845f
1
Parent(s): 0b2675d
fix: compute dynamic initial values for active p1 and available units on reset observation
Browse files
src/openenv_environment.py
CHANGED
|
@@ -23,6 +23,13 @@ class OpenEnvEnvironment:
|
|
| 23 |
self._state.metadata["cumulative_reward"] = 0.0
|
| 24 |
self._state.metadata["episode_rewards"] = []
|
| 25 |
self._state.metadata["episode_score"] = 0.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
self._last_observation = Observation(
|
| 27 |
result="dispatch center online",
|
| 28 |
score=0.0,
|
|
@@ -36,6 +43,8 @@ class OpenEnvEnvironment:
|
|
| 36 |
"protocol": 1.0,
|
| 37 |
},
|
| 38 |
phraseology_score=1.0,
|
|
|
|
|
|
|
| 39 |
)
|
| 40 |
return self._last_observation
|
| 41 |
|
|
|
|
| 23 |
self._state.metadata["cumulative_reward"] = 0.0
|
| 24 |
self._state.metadata["episode_rewards"] = []
|
| 25 |
self._state.metadata["episode_score"] = 0.0
|
| 26 |
+
active_p1 = sum(
|
| 27 |
+
1
|
| 28 |
+
for i in self._state.incidents.values()
|
| 29 |
+
if i.severity.value == "PRIORITY_1" and i.status.value not in {"RESOLVED", "ESCALATED"}
|
| 30 |
+
)
|
| 31 |
+
avail = sum(1 for u in self._state.units.values() if u.status.value == "AVAILABLE")
|
| 32 |
+
|
| 33 |
self._last_observation = Observation(
|
| 34 |
result="dispatch center online",
|
| 35 |
score=0.0,
|
|
|
|
| 43 |
"protocol": 1.0,
|
| 44 |
},
|
| 45 |
phraseology_score=1.0,
|
| 46 |
+
active_p1_count=active_p1,
|
| 47 |
+
units_available=avail,
|
| 48 |
)
|
| 49 |
return self._last_observation
|
| 50 |
|