Update tutorials

Author: kburns

dedalus/core/distributor.py

@@ -296,7 +296,7 @@ def local_grids(self, *bases, scales=None):
 
     def local_modes(self, basis):
         # TODO: remove from bases and do it all here?
-        return basis.local_modes()
+        return basis.local_modes(self)
 
     @CachedAttribute
     def default_nonconst_groups(self):

dedalus/extras/plot_tools.py

@@ -34,6 +34,7 @@ def __init__(self, field, axis):
         self.field = field
         self.axis = axis
         self.basis = field.domain.bases[axis]
+        self.dist = field.dist
 
     @property
     def label(self):
@@ -45,9 +46,9 @@ def label(self):
     def __getitem__(self, item):
         if self.field.layout.grid_space[self.axis]:
             scale = self.field.scales[self.axis]
-            return self.basis.global_grid(scale).ravel()
+            return self.basis.global_grid(self.dist, scale).ravel()
         else:
-            return self.basis.local_modes().ravel()
+            return self.dist.local_modes(self.basis).ravel()
 
 
 def plot_bot(dset, image_axes, data_slices, image_scales=(0,0), clim=None, even_scale=False, cmap='RdBu_r', axes=None, figkw={}, title=None, func=None, visible_axes=True):