From b3d4eb4ec0920201291daa8c1feecc17ffb5baca Mon Sep 17 00:00:00 2001 From: Marek Wolan Date: Wed, 5 Jul 2023 13:58:46 +0100 Subject: [PATCH] Changed hardcoded agent helper for new obs space --- src/primaite/agents/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/primaite/agents/utils.py b/src/primaite/agents/utils.py index 8c59faf7..b5a3c673 100644 --- a/src/primaite/agents/utils.py +++ b/src/primaite/agents/utils.py @@ -165,7 +165,8 @@ def transform_change_obs_readable(obs): os_states = [SoftwareState(i).name for i in obs[:, 2]] 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 service_states = [SoftwareState(i).name if i <= 4 else i for i in obs[:, service]] new_obs.append(service_states)