Added spaces to the checkpoint_read function

geodynamics · Jun 21, 2024 · 637065e · 637065e
1 parent 9a03f1f
commit 637065e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/post_processing/rayleigh_diagnostics.py b/post_processing/rayleigh_diagnostics.py
@@ -3267,8 +3267,8 @@ def streamfunction(vr,vt,r,cost,order=0):
     return psi
 
 ###### Function for reading in the checkpoint files (i.e. P,PAB,T,TAB,W,WAB,Z,ZAB)
-def checkpoint_read(chk_file_string,nr,ntheta):
-    nell=(2*ntheta)//3
+def checkpoint_read(chk_file_string, nr, ntheta):
+    nell = (2*ntheta)//3
     shape = (nell,nell)
     i,j = np.indices(shape)
     m = i <= j