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

updated stuff

Browse Source
This commit is contained in:
romue
2021-05-07 14:01:59 +02:00
parent 6eb97e20b1
commit 3c86684586
2 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
environments/helpers.py
View File

@ -28,6 +28,7 @@ def check_agent_move(state, dim, action):
raise AssertionError('Only one agent per slice is allowed.')
x, y = agent_pos[0]
x_new, y_new = x, y
# Actions
if action == 0: # North
x_new -= 1
elif action == 1: # East
@ -48,6 +49,14 @@ def check_agent_move(state, dim, action):
elif action == 7: # NW
x_new -= 1
y_new -= 1
# Check validity
valid = (x_new < 0 or y_new < 0
or x_new >= agent_slice.shape[0]
or y_new >= agent_slice.shape[0]
)
return (x, y), (x_new, y_new), valid