Skip to content

Commit

Permalink
examples: fix flake8
Browse files Browse the repository at this point in the history
  • Loading branch information
ofmla committed Jul 27, 2023
1 parent f83981b commit ff12414
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 7 deletions.
11 changes: 5 additions & 6 deletions examples/seismic/tti/operators.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,7 @@
from devito import (Eq, Operator, Function, TimeFunction, NODE, Inc, solve,
cos, sin, sqrt, sign)
cos, sin, sqrt)
from examples.seismic import PointSource, Receiver
from examples.seismic.acoustic.operators import freesurface
from devito.symbolics import retrieve_functions, INT


def second_order_stencil(model, u, v, H0, Hz, qu, qv, forward=True):
Expand All @@ -26,12 +25,12 @@ def second_order_stencil(model, u, v, H0, Hz, qu, qv, forward=True):
second_stencil = Eq(vnext, stencilr, subdomain=model.grid.subdomains['physdomain'])

stencils = [first_stencil, second_stencil]

# Add free surface
if model.fs:
stencils.append(freesurface(model, Eq(unext, stencilp)))
stencils.append(freesurface(model, Eq(vnext, stencilr)))
stencils.append(freesurface(model, Eq(unext, stencilp)))
stencils.append(freesurface(model, Eq(vnext, stencilr)))

return stencils


Expand Down
2 changes: 1 addition & 1 deletion examples/seismic/tti/wavesolver.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def __init__(self, model, geometry, space_order=4, kernel='centered',
self.model._initialize_bcs(bcs="damp")
self.geometry = geometry
self.kernel = kernel

if model.fs and kernel == 'staggered':
warning("Bad arguments in AnisotropicWaveSolver() - " +
"if model.fs is True, kernel should be centered. " +
Expand Down

0 comments on commit ff12414

Please sign in to comment.