From 83e95c4654d4a7d7bc1e711f6622c31b1f1c1561 Mon Sep 17 00:00:00 2001
From: Eduardo Moscatelli de Souza <5752216+SouzaEM@users.noreply.github.com>
Date: Fri, 1 Nov 2024 14:28:43 -0300
Subject: [PATCH] Lint modifications from issue 130

---
 spyro/examples/elastic_local_abc.py       | 4 ++--
 spyro/io/field_logger.py                  | 2 +-
 spyro/solvers/elastic_wave/functionals.py | 2 +-
 spyro/solvers/elastic_wave/local_abc.py   | 2 ++
 test/test_elastic_local_abc.py            | 4 ++--
 5 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/spyro/examples/elastic_local_abc.py b/spyro/examples/elastic_local_abc.py
index a403b4c9..379da327 100644
--- a/spyro/examples/elastic_local_abc.py
+++ b/spyro/examples/elastic_local_abc.py
@@ -1,4 +1,3 @@
-import firedrake as fire
 import numpy as np
 import spyro
 
@@ -21,6 +20,7 @@
 final_time = 2.0  # [s]
 out_freq = int(0.01/time_step)
 
+
 def build_solver(local_abc, dt_scheme):
     d = {}
 
@@ -91,4 +91,4 @@ def build_solver(local_abc, dt_scheme):
     wave = spyro.IsotropicWave(d)
     wave.set_mesh(mesh_parameters={'dx': h})
 
-    return wave
\ No newline at end of file
+    return wave
diff --git a/spyro/io/field_logger.py b/spyro/io/field_logger.py
index 4194bdb7..358dbe88 100644
--- a/spyro/io/field_logger.py
+++ b/spyro/io/field_logger.py
@@ -107,4 +107,4 @@ def log(self, t):
                 functional.sample()
 
     def get(self, key):
-        return self.__enabled_functionals[key].list[-1]
\ No newline at end of file
+        return self.__enabled_functionals[key].list[-1]
diff --git a/spyro/solvers/elastic_wave/functionals.py b/spyro/solvers/elastic_wave/functionals.py
index 5f90c6ce..e517b284 100644
--- a/spyro/solvers/elastic_wave/functionals.py
+++ b/spyro/solvers/elastic_wave/functionals.py
@@ -18,4 +18,4 @@ def mechanical_energy_form(wave):
     eps = lambda v: 0.5*(grad(v) + grad(v).T)
     U = (lmbda*div(u_n)*div(u_n) + 2*mu*inner(eps(u_n), eps(u_n)))*dx
 
-    return K + U
\ No newline at end of file
+    return K + U
diff --git a/spyro/solvers/elastic_wave/local_abc.py b/spyro/solvers/elastic_wave/local_abc.py
index 86bfb98e..de515113 100644
--- a/spyro/solvers/elastic_wave/local_abc.py
+++ b/spyro/solvers/elastic_wave/local_abc.py
@@ -86,6 +86,7 @@ def local_abc_form(wave):
                     uz_dx, ux_dx, uy_dx,
                     uz_dy, ux_dy, uy_dy)
 
+
 def clayton_engquist_A1_terms(ndim, rho, c_p, c_s,
                               v, iz, ix, iy, qr_s,
                               uz_dt, ux_dt, uy_dt,
@@ -150,6 +151,7 @@ def clayton_engquist_A1_terms(ndim, rho, c_p, c_s,
 
     return F_t
 
+
 def stacey_terms(ndim, rho, c_p, c_s,
                  v, iz, ix, iy, qr_s,
                  uz_dt, ux_dt, uy_dt,
diff --git a/test/test_elastic_local_abc.py b/test/test_elastic_local_abc.py
index 7ff1a416..724b7896 100644
--- a/test/test_elastic_local_abc.py
+++ b/test/test_elastic_local_abc.py
@@ -22,7 +22,7 @@ def test_clayton_engquist_abc():
 
 def test_with_central():
     wave = build_solver("Stacey", "central")
-    with pytest.raises(AssertionError) as e:
+    with pytest.raises(AssertionError):
         wave.forward_solve()
 
 
@@ -30,4 +30,4 @@ def test_with_backward_2nd():
     wave = build_solver("Stacey", "backward_2nd")
     wave.forward_solve()
     last_mechanical_energy = wave.field_logger.get("mechanical_energy")
-    assert last_mechanical_energy < expected_mechanical_energy
\ No newline at end of file
+    assert last_mechanical_energy < expected_mechanical_energy