From ff12414b36d38e0d7e1b0a441cd20c7e00c20389 Mon Sep 17 00:00:00 2001 From: ofmla Date: Wed, 26 Jul 2023 15:22:49 -0300 Subject: [PATCH] examples: fix flake8 --- examples/seismic/tti/operators.py | 11 +++++------ examples/seismic/tti/wavesolver.py | 2 +- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/examples/seismic/tti/operators.py b/examples/seismic/tti/operators.py index b30cf214c90..b6e8687c041 100644 --- a/examples/seismic/tti/operators.py +++ b/examples/seismic/tti/operators.py @@ -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): @@ -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 diff --git a/examples/seismic/tti/wavesolver.py b/examples/seismic/tti/wavesolver.py index 9b7b56451e3..aa51665ac9a 100644 --- a/examples/seismic/tti/wavesolver.py +++ b/examples/seismic/tti/wavesolver.py @@ -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. " +