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

Changed hardcoded agent helper for new obs space

Browse Source
This commit is contained in:
Marek Wolan
2023-07-05 13:58:46 +01:00
parent e199dc52c0
commit b3d4eb4ec0
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/primaite/agents/utils.py
View File

@@ -165,7 +165,8 @@ def transform_change_obs_readable(obs):
os_states = [SoftwareState(i).name for i in obs[:, 2]] os_states = [SoftwareState(i).name for i in obs[:, 2]]
new_obs = [ids, operating_states, os_states] new_obs = [ids, operating_states, os_states]
for service in range(3, obs.shape[1]): # changed range(3,...) to range(4,...) because we added file system which was new since ADSP
for service in range(4, obs.shape[1]):
# Links bit/s don't have a service state # Links bit/s don't have a service state
service_states = [SoftwareState(i).name if i <= 4 else i for i in obs[:, service]] service_states = [SoftwareState(i).name if i <= 4 else i for i in obs[:, service]]
new_obs.append(service_states) new_obs.append(service_states)