style: pre-commit fixes

Author: pre-commit-ci[bot]

pybamm/solvers/idaklu_solver.py

@@ -533,7 +533,7 @@ def _integrate(self, model, t_eval, inputs_dict=None):
 
         # return sensitivity solution, we need to flatten yS to
         # (#timesteps * #states (where t is changing the quickest),) to match format used by Solution
-        # note that yS is (n_p, n_t, n_y)  
+        # note that yS is (n_p, n_t, n_y)
         if number_of_sensitivity_parameters != 0:
             yS_out = {
                 name: sol.yS[i].reshape(-1, 1)
@@ -543,7 +543,7 @@ def _integrate(self, model, t_eval, inputs_dict=None):
             yS_out["all"] = np.hstack([yS_out[name] for name in sensitivity_names])
         else:
             yS_out = False
-            
+
         if sol.flag in [0, 2]:
             # 0 = solved for all t_eval
             if sol.flag == 0:

tests/unit/test_solvers/test_idaklu_solver.py

@@ -251,11 +251,10 @@ def test_ida_roberts_klu_sensitivities(self):
             dyda_fd = dyda_fd.transpose().reshape(-1, 1)
 
             np.testing.assert_array_almost_equal(dyda_ida, dyda_fd)
-            
+
             # get the sensitivities for the variable
             d2uda = sol["2u"].sensitivities["a"]
             np.testing.assert_array_almost_equal(2 * dyda_ida[0:200:2], d2uda)
-            
 
     def test_sensitivities_with_events(self):
         # this test implements a python version of the ida Roberts