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

#2658: fix space

Browse Source
This commit is contained in:
Czar Echavez
2024-06-13 12:37:20 +01:00
parent 539577ddc3
commit 1798674d39
1 changed files with 16 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
src/primaite/game/agent/observations/nic_observations.py
View File

@@ -63,6 +63,7 @@ class NICObservation(AbstractObservation, identifier="NETWORK_INTERFACE"):
if protocol == "icmp": if protocol == "icmp":
default_traffic_obs["monitored_traffic"]["icmp"] = {"inbound": 0, "outbound": 0} default_traffic_obs["monitored_traffic"]["icmp"] = {"inbound": 0, "outbound": 0}
else: else:
default_traffic_obs["monitored_traffic"][protocol] = {}
for port in monitored_traffic_config[protocol]: for port in monitored_traffic_config[protocol]:
default_traffic_obs["monitored_traffic"][protocol][Port[port].value] = {"inbound": 0, "outbound": 0} default_traffic_obs["monitored_traffic"][protocol][Port[port].value] = {"inbound": 0, "outbound": 0}
@@ -186,7 +187,21 @@ class NICObservation(AbstractObservation, identifier="NETWORK_INTERFACE"):
if self.include_nmne: if self.include_nmne:
space["NMNE"] = spaces.Dict({"inbound": spaces.Discrete(4), "outbound": spaces.Discrete(4)}) space["NMNE"] = spaces.Dict({"inbound": spaces.Discrete(4), "outbound": spaces.Discrete(4)})
return space if self.monitored_traffic:
space["monitored_traffic"] = spaces.Dict({})
for protocol in self.monitored_traffic:
if protocol == "icmp":
space["monitored_traffic"]["icmp"] = spaces.Dict(
{"inbound": spaces.Discrete(11), "outbound": spaces.Discrete(11)}
)
else:
space["monitored_traffic"][protocol] = spaces.Dict({})
for port in self.monitored_traffic[protocol]:
space["monitored_traffic"][protocol][Port[port].value] = spaces.Dict(
{"inbound": spaces.Discrete(11), "outbound": spaces.Discrete(11)}
)
return spaces.Dict(space)
@classmethod @classmethod
def from_config(cls, config: ConfigSchema, parent_where: WhereType = []) -> NICObservation: def from_config(cls, config: ConfigSchema, parent_where: WhereType = []) -> NICObservation: