is debugged, no longer dirty at [2, 8]

2021-05-18 10:35:03 +02:00
parent 8769bc8d7b
commit 2acf91b395
2 changed files with 2 additions and 8 deletions
--- a/environments/factory/base_factory.py
+++ b/environments/factory/base_factory.py
@@ -160,9 +160,7 @@ class BaseFactory:
            excluded_slices = [inds[x] if x < 0 else x for x in excluded_slices]
            state = self.state[[x for x in inds if x not in excluded_slices]]

-        free_cells = state.sum(0)
-        free_cells[excluded_slices] = 0
-        free_cells = np.argwhere(free_cells == h.IS_FREE_CELL)
+        free_cells = np.argwhere(state.sum(0) == h.IS_FREE_CELL)
        np.random.shuffle(free_cells)
        return free_cells