SayedZahur786 commited on
Commit
aea845f
·
1 Parent(s): 0b2675d

fix: compute dynamic initial values for active p1 and available units on reset observation

Browse files
Files changed (1) hide show
  1. src/openenv_environment.py +9 -0
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