Fix multi-channel generation

BerkeleyAutomation · Jul 30, 2018 · 449d1cd · 449d1cd
1 parent 7a7c0f5
commit 449d1cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fluids/obs/grid.py b/fluids/obs/grid.py
@@ -140,5 +140,5 @@ def render(self, surface):
     def get_array(self):
         arr = np.zeros((self.grid_dim, self.grid_dim, len(self.pygame_rep)))
         for i in range(len(self.pygame_rep)):
-            arr[:,:,i] = pygame.surfarray.array2d(self.pygame_rep[0]) > 0
+            arr[:,:,i] = pygame.surfarray.array2d(self.pygame_rep[i]) > 0
         return arr