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

Remove redundant cols from transactions

Browse Source
This commit is contained in:
Marek Wolan
2023-06-30 10:41:56 +01:00
parent fb48f75adf
commit 99ba05c6ee
4 changed files with 8 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/primaite/environment/observations.py
View File

@@ -168,7 +168,7 @@ class NodeLinkTable(AbstractObservationComponent):
f"link_{link_id}_n/a",
]
for j, serv in enumerate(self.env.services_list):
link_labels.append(f"node_{node_id}_service_{serv}_load")
link_labels.append(f"link_{link_id}_service_{serv}_load")
structure.extend(link_labels)
return structure

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

@@ -318,7 +318,7 @@ class Primaite(Env):
datetime.now(), self.agent_identifier, self.episode_count, self.step_count
)
# Load the initial observation space into the transaction
transaction.set_obs_space_pre(self.obs_handler._flat_observation)
transaction.set_obs_space(self.obs_handler._flat_observation)
# Load the action space into the transaction
transaction.set_action_space(copy.deepcopy(action))
@@ -400,8 +400,6 @@ class Primaite(Env):
# 7. Update env_obs
self.update_environent_obs()
# Load the new observation space into the transaction
transaction.set_obs_space_post(self.obs_handler._flat_observation)
# 8. Add the transaction to the list of transactions
self.transaction_list.append(copy.deepcopy(transaction))