Change software describe state keys

2024-01-04 12:55:46 +00:00
parent ffeb04d6ed
commit bc367222a8
2 changed files with 6 additions and 3 deletions
--- a/src/primaite/game/agent/observations.py
+++ b/src/primaite/game/agent/observations.py
@@ -139,7 +139,10 @@ class ServiceObservation(AbstractObservation):
        service_state = access_from_nested_dict(state, self.where)
        if service_state is NOT_PRESENT_IN_STATE:
            return self.default_observation
-        return {"operating_status": service_state["operating_state"], "health_status": service_state["health_state"]}
+        return {
+            "operating_status": service_state["operating_state"],
+            "health_status": service_state["health_state_visible"],
+        }

    @property
    def space(self) -> spaces.Space: