diff --git a/environments/factory/assets/agents/agent_collision.png b/environments/factory/assets/agents/agent_collision.png new file mode 100644 index 0000000..d7f228c Binary files /dev/null and b/environments/factory/assets/agents/agent_collision.png differ diff --git a/environments/factory/simple_factory_getting_dirty.py b/environments/factory/simple_factory_getting_dirty.py index 39dd988..2caf0ba 100644 --- a/environments/factory/simple_factory_getting_dirty.py +++ b/environments/factory/simple_factory_getting_dirty.py @@ -45,9 +45,9 @@ class GettingDirty(BaseFactory): def asset_str(agent): cols = ' '.join([self.slice_strings[j] for j in agent.collisions]) - asset_str = f'agent{agent.i + 1}violation' if (not agent.action_valid or 'level' in cols or 'agent' in cols) \ + if 'agent' in cols: return 'agent_collision' + return f'agent{agent.i + 1}violation' if (not agent.action_valid or 'level' in cols or 'agent' in cols) \ else (f'agent{agent.i + 1}valid' if self._is_clean_up_action(agent.action) else f'agent{agent.i + 1}') - return asset_str agents = {f'agent{i+1}': [Entity(asset_str(agent), agent.pos)] for i, agent in enumerate(self.agent_states)}