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

fixed movement bug

Browse Source
This commit is contained in:
romue
2021-05-07 14:51:10 +02:00
parent 83d399901b
commit e993e8642a
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
environments/factory/factory_cleaning.py
View File

@ -40,17 +40,17 @@ class Factory(object):
for i in range(self.n_agents): # might as well save the positions (redundant)
agent_slice = self.state[i+1]
x, y = np.argwhere(agent_slice == 1)[0]
collisions_vec = self.state[:, x, y]
collisions_vec = self.state[:, x, y].copy() # otherwise you overwrite the grid/state
collisions_vec[i+1] = 0 # no self-collisions
collision_vecs.append(collisions_vec)
self.handle_collisions(collisions_vec)
return self.state, r, self.done, {}
def make_move(self, agent_i, old_pos, new_pos):
(x, y), (x_new, y_new) = old_pos, new_pos
self.state[agent_i, x, y] = 0
self.state[agent_i, x_new, y_new] = 1
self.state[agent_i+1, x, y] = 0
self.state[agent_i+1, x_new, y_new] = 1
print( old_pos, new_pos)
def handle_collisions(self, vecs):
pass
@ -58,5 +58,6 @@ class Factory(object):
if __name__ == '__main__':
factory = Factory(n_agents=1)
factory.step(0)
print(factory.state.shape)
print(factory.state)
state, r, done, _ = factory.step(0)
print(state)