diff --git a/src/primaite/game/agent/observations.py b/src/primaite/game/agent/observations.py index e5216e4a..cac5b91e 100644 --- a/src/primaite/game/agent/observations.py +++ b/src/primaite/game/agent/observations.py @@ -79,7 +79,7 @@ class FileObservation(AbstractObservation): file_state = access_from_nested_dict(state, self.where) if file_state is NOT_PRESENT_IN_STATE: return self.default_observation - return {"health_status": file_state["health_status"]} + return {"health_status": file_state["visible_status"]} @property def space(self) -> spaces.Space: diff --git a/src/primaite/game/agent/rewards.py b/src/primaite/game/agent/rewards.py index 30baad6f..8f064be3 100644 --- a/src/primaite/game/agent/rewards.py +++ b/src/primaite/game/agent/rewards.py @@ -111,7 +111,7 @@ 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 == 3: + if health_status == 2: return -1 elif health_status == 1: return 1 diff --git a/src/primaite/simulator/file_system/folder.py b/src/primaite/simulator/file_system/folder.py index d4e72f63..237a6341 100644 --- a/src/primaite/simulator/file_system/folder.py +++ b/src/primaite/simulator/file_system/folder.py @@ -105,7 +105,7 @@ class Folder(FileSystemItemABC): self._file_request_manager = RequestManager() rm.add_request( name="file", - request_type=RequestType(func=lambda request, context: self._file_request_manager), + request_type=RequestType(func=self._file_request_manager), ) return rm