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

893 - added test which shows the new action space has been created when ANY is selected in single_action_space_lay_down_config.yaml

Browse Source
This commit is contained in:
SunilSamra
2023-06-02 11:55:31 +01:00
parent d0c11a14ed
commit 66fdae5df1
2 changed files with 19 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/primaite/environment/primaite_env.py
View File

@@ -249,11 +249,13 @@ class Primaite(Env):
self.action_dict = self.create_acl_action_dict()
self.action_space = spaces.Discrete(len(self.action_dict))
print(self.action_space, "ACL action space")
else:
_LOGGER.warning("Action space type ANY selected - Node + ACL")
elif self.action_type == ActionType.ANY:
_LOGGER.info("Action space type ANY selected - Node + ACL")
self.action_dict = self.create_node_and_acl_action_dict()
self.action_space = spaces.Discrete(len(self.action_dict))
print(self.action_space, "ANY action space")
else:
_LOGGER.info("Invalid action type selected")
# Set up a csv to store the results of the training
try:
now = datetime.now() # current date and time
@@ -1271,6 +1273,5 @@ class Primaite(Env):
# Combine the Node dict and ACL dict
combined_action_dict = {**acl_action_dict, **new_node_action_dict}
logging.warning("logging is working")
# print(len(list(combined_action_dict.values())))
return combined_action_dict