Skip to content

Commit b2d1d4f

Browse files
authored
1 parent 1e738f3 commit b2d1d4f

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

examples/seismic/plotting.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ def plot_perturbation(model, model1, colorbar=True):
2929
domain_size = 1.e-3 * np.array(model.domain_size)
3030
extent = [model.origin[0], model.origin[0] + domain_size[0],
3131
model.origin[1] + domain_size[1], model.origin[1]]
32+
assert model.nbl == model1.nbl, ValueError("model and model1 have different values for nbl")
3233
slices = tuple(slice(model.nbl, -model.nbl) for _ in range(2))
33-
slices1 = tuple(slice(model1.nbl, -model1.nbl) for _ in range(2))
34-
dv = np.transpose(model.vp.data[slices]) - np.transpose(model1.vp.data[slices1])
34+
dv = np.transpose(model.vp.data[slices]) - np.transpose(model1.vp.data[slices])
3535

3636
plot = plt.imshow(dv, animated=True, cmap=cm.jet,
3737
vmin=min(dv.reshape(-1)), vmax=max(dv.reshape(-1)),

0 commit comments

Comments
 (0)