#2913: Fix remaining test errors.

This commit is contained in:
Nick Todd
2024-10-22 12:14:30 +01:00
parent f95ba8cbbc
commit 318f8926f0

View File

@@ -272,7 +272,6 @@ class WebpageUnavailablePenalty(AbstractReward, identifier="WebpageUnavailablePe
sticky: bool = True sticky: bool = True
reward: float = 0.0 reward: float = 0.0
location_in_state: List[str] = [""] location_in_state: List[str] = [""]
_node: str = node_hostname
class ConfigSchema(AbstractReward.ConfigSchema): class ConfigSchema(AbstractReward.ConfigSchema):
"""ConfigSchema for WebpageUnavailablePenalty.""" """ConfigSchema for WebpageUnavailablePenalty."""
@@ -311,7 +310,7 @@ class WebpageUnavailablePenalty(AbstractReward, identifier="WebpageUnavailablePe
request_attempted = last_action_response.request == [ request_attempted = last_action_response.request == [
"network", "network",
"node", "node",
self._node, self.node_hostname,
"application", "application",
"WebBrowser", "WebBrowser",
"execute", "execute",
@@ -326,7 +325,7 @@ class WebpageUnavailablePenalty(AbstractReward, identifier="WebpageUnavailablePe
elif web_browser_state is NOT_PRESENT_IN_STATE or not web_browser_state["history"]: elif web_browser_state is NOT_PRESENT_IN_STATE or not web_browser_state["history"]:
_LOGGER.debug( _LOGGER.debug(
"Web browser reward could not be calculated because the web browser history on node", "Web browser reward could not be calculated because the web browser history on node",
f"{self._node} was not reported in the simulation state. Returning 0.0", f"{self.node_hostname} was not reported in the simulation state. Returning 0.0",
) )
self.reward = 0.0 self.reward = 0.0
else: else:
@@ -357,7 +356,6 @@ class GreenAdminDatabaseUnreachablePenalty(AbstractReward, identifier="GreenAdmi
"""Penalises the agent when the green db clients fail to connect to the database.""" """Penalises the agent when the green db clients fail to connect to the database."""
node_hostname: str = "" node_hostname: str = ""
_node: str = node_hostname
sticky: bool = True sticky: bool = True
reward: float = 0.0 reward: float = 0.0
@@ -385,7 +383,7 @@ class GreenAdminDatabaseUnreachablePenalty(AbstractReward, identifier="GreenAdmi
request_attempted = last_action_response.request == [ request_attempted = last_action_response.request == [
"network", "network",
"node", "node",
self._node, self.node_hostname,
"application", "application",
"DatabaseClient", "DatabaseClient",
"execute", "execute",