Merged PR 257: Fix reward data type

Related work items: #2179
2024-01-11 11:07:01 +00:00
parent b11e4c8ccd 4d1c0d268e
commit 40ad446ff2
1 changed files with 2 additions and 2 deletions
--- a/src/primaite/game/agent/rewards.py
+++ b/src/primaite/game/agent/rewards.py
@@ -111,9 +111,9 @@ class DatabaseFileIntegrity(AbstractReward):
        """
        database_file_state = access_from_nested_dict(state, self.location_in_state)
        health_status = database_file_state["health_status"]
-        if health_status == "corrupted":
+        if health_status == 3:
            return -1
-        elif health_status == "good":
+        elif health_status == 1:
            return 1
        else:
            return 0