style: pre-commit fixes

pybamm/solvers/processed_variable.py

@@ -366,7 +366,9 @@ def _process_spatial_variable_names(self, spatial_variable):
                 f"Spatial variable name not recognized for {spatial_variable}"
             )
 
-    def __call__(self, t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **kwargs):
+    def __call__(
+        self, t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **kwargs
+    ):
         """
         Evaluate the variable at arbitrary *dimensional* t (and x, r, y, z and/or R),
         using interpolation
@@ -376,12 +378,13 @@ def __call__(self, t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **
         spatial_vars.update(kwargs)
 
         # Remove any None arguments
-        spatial_vars = {key: value for key, value in spatial_vars.items() if value is not None}
+        spatial_vars = {
+            key: value for key, value in spatial_vars.items() if value is not None
+        }
 
         # Use xarray interpolation, return numpy array
         return self._xr_data_array.interp(**spatial_vars).values
 
-
     @property
     def data(self):
         """Same as entries, but different name"""

pybamm/solvers/processed_variable_computed.py

@@ -411,7 +411,9 @@ def initialise_2D_scikit_fem(self):
             coords={"y": y_sol, "z": z_sol, "t": self.t_pts},
         )
 
-    def __call__(self, t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **kwargs):
+    def __call__(
+        self, t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **kwargs
+    ):
         """
         Evaluate the variable at arbitrary *dimensional* t (and x, r, y, z and/or R),
         using interpolation
@@ -421,7 +423,9 @@ def __call__(self, t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **
         spatial_vars.update(kwargs)
 
         # Remove any None arguments
-        spatial_vars = {key: value for key, value in spatial_vars.items() if value is not None}
+        spatial_vars = {
+            key: value for key, value in spatial_vars.items() if value is not None
+        }
 
         # Use xarray interpolation, return numpy array
         return self._xr_data_array.interp(**spatial_vars).values

tests/unit/test_solvers/test_processed_variable.py

@@ -58,9 +58,11 @@ def process_and_check_2D_variable(
     )
     return y_sol, first_sol, second_sol, t_sol
 
+
 def call_func(t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **kwargs):
     return np.random.rand(5, 5)
 
+
 class TestProcessedVariable(TestCase):
     def test_processed_variable_0D(self):
         # without space
@@ -1162,7 +1164,7 @@ def test_call_with_R(self):
         self.assertIsInstance(result, np.ndarray)
 
     def test_call_with_kwargs(self):
-        result = call_func(foo='bar')
+        result = call_func(foo="bar")
         self.assertIsInstance(result, np.ndarray)
 
 

tests/unit/test_solvers/test_processed_variable_computed.py

@@ -65,9 +65,11 @@ def process_and_check_2D_variable(
     #    ::test_output_variables
     return y_sol, first_sol, second_sol, t_sol
 
+
 def call_func(t=None, x=None, r=None, y=None, z=None, R=None, warn=True, **kwargs):
     return np.random.rand(5, 5)
 
+
 class TestProcessedVariableComputed(TestCase):
     def test_processed_variable_0D(self):
         # without space
@@ -472,9 +474,10 @@ def test_call_with_R(self):
         self.assertIsInstance(result, np.ndarray)
 
     def test_call_with_kwargs(self):
-        result = call_func(foo='bar')
+        result = call_func(foo="bar")
         self.assertIsInstance(result, np.ndarray)
 
+
 if __name__ == "__main__":
     print("Add -v for more debug output")
     import sys