check if pcolormesh data is a list

t-makaro · Mar 5, 2019 · 27277ea · 27277ea
1 parent bcda8f5
commit 27277ea
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/animatplot/blocks/image_like.py b/animatplot/blocks/image_like.py
@@ -60,6 +60,8 @@ def _update(self, i):
         return self.quad
 
     def __len__(self):
+        if self._is_list:
+            return self.C.shape[0]
         return self.C.shape[self.t_axis]
 
     def _make_pcolormesh_slice(self, i, dim):