Update tutorials

DedalusProject · Jan 8, 2024 · c5ae610 · c5ae610
1 parent 68dcb74
commit c5ae610
Show file tree

Hide file tree

Showing 6 changed files with 89 additions and 95 deletions.
diff --git a/dedalus/core/distributor.py b/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):

diff --git a/dedalus/extras/plot_tools.py b/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):

diff --git a/docs/notebooks/dedalus_tutorial_1.ipynb b/docs/notebooks/dedalus_tutorial_1.ipynb
diff --git a/docs/notebooks/dedalus_tutorial_2.ipynb b/docs/notebooks/dedalus_tutorial_2.ipynb
diff --git a/docs/notebooks/dedalus_tutorial_3.ipynb b/docs/notebooks/dedalus_tutorial_3.ipynb
diff --git a/docs/notebooks/dedalus_tutorial_4.ipynb b/docs/notebooks/dedalus_tutorial_4.ipynb