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

updated base factory

Browse Source
This commit is contained in:
romue
2021-05-10 10:50:45 +02:00
parent ed10b3225d
commit a4a4ab067e
2 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
environments/factory/base_factory.py
View File

@ -45,19 +45,16 @@ class BaseFactory(object):
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)
r += self.step_core(collisions_vec, actions, r)
reward, info = self.step_core(collisions_vec, actions, r)
r += reward
if self.steps >= self.max_steps:
self.done = True
return self.state, r, self.done, {}
return self.state, r, self.done, info
def make_move(self, agent_i, old_pos, new_pos):
(x, y), (x_new, y_new) = old_pos, new_pos
self.state[agent_i+1, x, y] = 0
self.state[agent_i+1, x_new, y_new] = 1
def handle_collisions(self, vecs):
pass
def step_core(self, collisions_vec, actions, r):
return 0
return 0, {}