#1100 flake8

pybamm-team · Jul 23, 2020 · 94314fb · 94314fb
1 parent 067c504
commit 94314fb
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 3 deletions.
diff --git a/pybamm/solvers/base_solver.py b/pybamm/solvers/base_solver.py
@@ -205,7 +205,7 @@ def set_up(self, model, inputs=None):
 
         # Only allow solving sensitivity equations with the casadi format for now
         if (
-            self.sensitivity is "explicit forward"
+            self.sensitivity == "explicit forward"
             and model.convert_to_format != "casadi"
         ):
             raise NotImplementedError(

diff --git a/pybamm/solvers/casadi_algebraic_solver.py b/pybamm/solvers/casadi_algebraic_solver.py
@@ -3,7 +3,6 @@
 #
 import casadi
 import pybamm
-import numbers
 import numpy as np
 
 

diff --git a/tests/unit/test_solvers/test_casadi_solver.py b/tests/unit/test_solvers/test_casadi_solver.py
@@ -6,7 +6,6 @@
 import numpy as np
 from tests import get_mesh_for_testing, get_discretisation_for_testing
 from scipy.sparse import eye
-from scipy.optimize import least_squares
 
 
 class TestCasadiSolver(unittest.TestCase):