diff --git a/environments/factory/assets/agents/agent1.png b/environments/factory/assets/agents/agent1.png deleted file mode 100644 index 0574988..0000000 Binary files a/environments/factory/assets/agents/agent1.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent1valid.png b/environments/factory/assets/agents/agent1valid.png deleted file mode 100644 index 1372803..0000000 Binary files a/environments/factory/assets/agents/agent1valid.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent1violation.png b/environments/factory/assets/agents/agent1violation.png deleted file mode 100644 index 119a0d3..0000000 Binary files a/environments/factory/assets/agents/agent1violation.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent2.png b/environments/factory/assets/agents/agent2.png deleted file mode 100644 index af03c8c..0000000 Binary files a/environments/factory/assets/agents/agent2.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent2valid.png b/environments/factory/assets/agents/agent2valid.png deleted file mode 100644 index d365287..0000000 Binary files a/environments/factory/assets/agents/agent2valid.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent2violation.png b/environments/factory/assets/agents/agent2violation.png deleted file mode 100644 index 90b8340..0000000 Binary files a/environments/factory/assets/agents/agent2violation.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent3.png b/environments/factory/assets/agents/agent3.png deleted file mode 100644 index e83b474..0000000 Binary files a/environments/factory/assets/agents/agent3.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent4.png b/environments/factory/assets/agents/agent4.png deleted file mode 100644 index 092560d..0000000 Binary files a/environments/factory/assets/agents/agent4.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent5.png b/environments/factory/assets/agents/agent5.png deleted file mode 100644 index d565aec..0000000 Binary files a/environments/factory/assets/agents/agent5.png and /dev/null differ diff --git a/environments/factory/assets/agents/agent6.png b/environments/factory/assets/agents/agent6.png deleted file mode 100644 index 36484dd..0000000 Binary files a/environments/factory/assets/agents/agent6.png and /dev/null differ diff --git a/environments/factory/simple_factory.py b/environments/factory/simple_factory.py index 21eb00b..1878830 100644 --- a/environments/factory/simple_factory.py +++ b/environments/factory/simple_factory.py @@ -59,11 +59,11 @@ class SimpleFactory(BaseFactory): if 'agent' in cols: return 'agent_collision', 'blank' elif not agent.action_valid or 'level' in cols or 'agent' in cols: - return f'agent{agent.i + 1}', 'invalid' + return 'agent', 'invalid' elif self._is_clean_up_action(agent.action): - return f'agent{agent.i + 1}', 'valid' + return 'agent', 'valid' else: - return f'agent{agent.i + 1}', 'idle' + return 'agent', 'idle' agents = [] for i, agent in enumerate(self._agent_states): name, state = asset_str(agent)