steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-06-20 03:08:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Refactoring of Collision checking bug fixes

Browse Source
This commit is contained in:
steffen-illium
2021-05-12 12:08:19 +02:00
parent c539e5dddd
commit 51d79d8ab3
3 changed files with 39 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
environments/factory/simple_factory_getting_dirty.py
View File

@ -30,7 +30,7 @@ class GettingDirty(BaseFactory):
self.state = np.concatenate((self.state, dirt_slice)) # dirt is now the last slice
self.spawn_dirt()
def step_core(self, collisions_vecs, actions, r):
def calculate_reward(self, collisions_vecs, actions, r):
for agent_i, cols in enumerate(collisions_vecs):
cols = np.argwhere(cols != 0).flatten()
print(f't = {self.steps}\tAgent {agent_i} has collisions with '