#1100 flake8

Author: valentinsulzer

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(

pybamm/solvers/casadi_algebraic_solver.py

@@ -3,7 +3,6 @@
 #
 import casadi
 import pybamm
-import numbers
 import numpy as np
 
 

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):