oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merged PR 257: Fix reward data type

Browse Source 
Related work items: #2179
This commit is contained in:
Marek Wolan
2024-01-11 11:07:01 +00:00
parent b11e4c8ccd 4d1c0d268e
commit 40ad446ff2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/primaite/game/agent/rewards.py
View File

@@ -111,9 +111,9 @@ class DatabaseFileIntegrity(AbstractReward):
""" """
database_file_state = access_from_nested_dict(state, self.location_in_state) database_file_state = access_from_nested_dict(state, self.location_in_state)
health_status = database_file_state["health_status"] health_status = database_file_state["health_status"]
if health_status == "corrupted": if health_status == 3:
return -1 return -1
elif health_status == "good": elif health_status == 1:
return 1 return 1
else: else:
return 0 return 0