fix validity check

environments/factory/factory_cleaning.py

@@ -32,6 +32,7 @@ class Factory(object):
         # level, agent 1,..., agent n,
         for i, a in enumerate(actions):
             old_pos, new_pos, valid = h.check_agent_move(state=self.state, dim=i+1, action=a)
+            print(old_pos, new_pos, valid)
 
 
 if __name__ == '__main__':

environments/helpers.py

@@ -50,10 +50,11 @@ def check_agent_move(state, dim, action):
         x_new -= 1
         y_new -= 1
     # Check validity
-    valid = (x_new < 0 or y_new < 0
+    valid = not (
+            x_new < 0 or y_new < 0
             or x_new >= agent_slice.shape[0]
             or y_new >= agent_slice.shape[0]
-             )
+    )  # if agent tried to leave the grid
     return (x, y), (x_new, y_new), valid