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

Minor fix

Browse Source
This commit is contained in:
Marek Wolan
2024-01-11 14:49:36 +00:00
parent 63b9bc5bc6
commit ed5591caf8
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/game/agent/observations.py
View File

@@ -79,7 +79,7 @@ class FileObservation(AbstractObservation):
file_state = access_from_nested_dict(state, self.where) file_state = access_from_nested_dict(state, self.where)
if file_state is NOT_PRESENT_IN_STATE: if file_state is NOT_PRESENT_IN_STATE:
return self.default_observation return self.default_observation
return {"health_status": file_state["health_status"]} return {"health_status": file_state["visible_status"]}
@property @property
def space(self) -> spaces.Space: def space(self) -> spaces.Space:

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

@@ -111,7 +111,7 @@ 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 == 3: if health_status == 2:
return -1 return -1
elif health_status == 1: elif health_status == 1:
return 1 return 1

2
src/primaite/simulator/file_system/folder.py
View File

@@ -105,7 +105,7 @@ class Folder(FileSystemItemABC):
self._file_request_manager = RequestManager() self._file_request_manager = RequestManager()
rm.add_request( rm.add_request(
name="file", name="file",
request_type=RequestType(func=lambda request, context: self._file_request_manager), request_type=RequestType(func=self._file_request_manager),
) )
return rm return rm