lint: Fix my linter

Author: JDBetteridge

devito/timestepping/superstep.py

@@ -70,10 +70,10 @@ def superstep_generator(field, stencil, k, tn=0):
 
 
 def superstep_solution_transfer(old, new, new_p, tn):
-    ''' Transfer the timesteps from a previous simulation to a 2 field superstep simulation
+    ''' Transfer state from a previous TimeFunction to a 2 field superstep
     Used after injecting source using standard timestepping.
     '''
-    idx = tn % 3  if old.save is None else -1
+    idx = tn % 3 if old.save is None else -1
     new.data[0, :] = old.data[idx - 1]
     new.data[1, :] = old.data[idx]
     new_p.data[0, :] = old.data[idx - 2]

examples/timestepping/acoustic_superstep_2d.py

@@ -52,9 +52,10 @@ def layered_velocity(grid, step=0):
     velocity.data[:, 3*q:] = 3000
     return velocity
 
+
 def marmousi(grid, step=0):
     # Grab dataset from pwd or download
-    url = 'https://github.com/devitocodes/data/raw/refs/heads/master/Simple2D/vp_marmousi_bi' # noqa: E501
+    url = 'https://github.com/devitocodes/data/raw/refs/heads/master/Simple2D/vp_marmousi_bi'  # noqa: E501
     filename = Path('marmousi.np')
     shape = (1601, 401)
     if not filename.exists():
@@ -95,7 +96,7 @@ def acoustic_model(model, step=1, snapshots=1):
     velocity = model.velocity(grid, step)
     u = TimeFunction(name="u", grid=grid, time_order=2, space_order=2)
 
-    pde = (1/velocity**2)*u.dt2 - u.laplace # + damp*u.dt
+    pde = (1/velocity**2)*u.dt2 - u.laplace
     stencil = Eq(u.forward, solve(pde, u.forward))
 
     tn1 = int(np.ceil((t1 - t0)/model.critical_dt))
@@ -158,29 +159,29 @@ def acoustic_model(model, step=1, snapshots=1):
 layered_model = model(
     name='layered',
     velocity=layered_velocity,
-    shape = (101, 101),
-    origin = (0., 0.),
-    extent = (1000, 1000), # 1kmx1km
-    source = (500, 20),
-    t0 = 0,
-    t1 = 0.2,
-    t2 = 0.65,
-    critical_dt = 0.002357,
-    zlim = 30
+    shape=(101, 101),
+    origin=(0., 0.),
+    extent=(1000, 1000),  # 1kmx1km
+    source=(500, 20),
+    t0=0,
+    t1=0.2,
+    t2=0.65,
+    critical_dt=0.002357,
+    zlim=30
 )
 
 marmousi_model = model(
     name='marmousi',
     velocity=marmousi,
-    shape = (274, 301),
-    origin = (4875., 262.5),
-    extent = (3000, 2737.5), # 3kmx2.7km
-    source = (1000, 1200),
-    t0 = 0,
-    t1 = 0.2,
-    t2 = 0.5,
-    critical_dt = 0.0013728,
-    zlim = 20
+    shape=(274, 301),
+    origin=(4875., 262.5),
+    extent=(3000, 2737.5),  # 3kmx2.7km
+    source=(1000, 1200),
+    t0=0,
+    t1=0.2,
+    t2=0.5,
+    critical_dt=0.0013728,
+    zlim=20
 )
 
 

examples/timestepping/superstep_1d.py

@@ -7,10 +7,10 @@
 from devito.timestepping.superstep import superstep_generator
 
 # Parameters
-## Spatial
+# Spatial
 shape = (501, )
 pad = (0, )
-origin= (0, )
+origin = (0, )
 extent = (1, )
 # Time
 t0 = 0
@@ -22,11 +22,13 @@
 ylim = np.ceil(1/np.sqrt(2*np.pi*sigma_sq))
 xlim = (0, 1)
 
+
 def gaussian(x, mu=0, sigma_sq=1):
     ''' Generate a Gaussian initial condition
     '''
     return np.exp(-((x - mu)**2)/(2*sigma_sq))/(np.sqrt(2*np.pi*sigma_sq))
 
+
 def wave_on_string(step=1):
     grid = Grid(shape=shape, extent=extent)
 
@@ -70,6 +72,7 @@ def wave_on_string(step=1):
     idx = tn % 3
     return newu.data[idx]
 
+
 if __name__ == '__main__':
     fig, ax = plt.subplots(1, 1)
     x = np.linspace(0, 1, *shape)

examples/timestepping/superstep_2d.py

@@ -10,10 +10,11 @@
 def gaussian2d(xx, yy, mu=0, sigma_sq=1):
     return np.exp(-((xx - mu)**2 + (yy - mu)**2)/(2*sigma_sq))/(np.sqrt(2*np.pi*sigma_sq))
 
+
 # Spatial Domain
 shape = (101, 101)
 origin = (0., 0.)
-extent = (1000, 1000) # 1kmx1km
+extent = (1000, 1000)  # 1kmx1km
 # Time Domain
 t0 = 0
 t1 = 0.5
@@ -35,8 +36,7 @@ def ripple_on_pond(step=1, snapshots=1):
 
     u = TimeFunction(name="u", grid=grid, time_order=2, space_order=2)
 
-    # The source/sink terms are injected in after the fact (???)
-    pde = (1/velocity**2)*u.dt2 - u.laplace # + damp*u.dt
+    pde = (1/velocity**2)*u.dt2 - u.laplace
     stencil = Eq(u.forward, solve(pde, u.forward))
 
     # Initial condition
@@ -82,6 +82,7 @@ def ripple_on_pond(step=1, snapshots=1):
 
     return u_save.data
 
+
 if __name__ == '__main__':
     # Supersteps
     k = [1, 3, 4]
@@ -111,4 +112,3 @@ def ripple_on_pond(step=1, snapshots=1):
         hspace=0.06
     )
     plt.show()
-