Fix MG failing test

firedrakeproject · Mar 18, 2021 · a42c063 · a42c063
1 parent 415ac82
commit a42c063
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/firedrake/assemble.py b/firedrake/assemble.py
@@ -172,8 +172,7 @@ def allocate_matrix(expr, bcs=(), form_compiler_parameters=None,
 
 
 def create_assembly_callable(expr, tensor=None, bcs=None, form_compiler_parameters=None,
-                             mat_type=None, sub_mat_type=None,
-                             diagonal=False):
+                             mat_type=None, sub_mat_type=None, diagonal=False):
     r"""Create a callable object than be used to assemble expr into a tensor.
 
     This is really only designed to be used inside residual and
@@ -186,10 +185,13 @@ def create_assembly_callable(expr, tensor=None, bcs=None, form_compiler_paramete
     """
     if tensor is None:
         raise ValueError("Have to provide tensor to write to")
-    return functools.partial(assemble, expr, tensor, bcs,
+    return functools.partial(assemble, expr,
+                             tensor=tensor,
+                             bcs=bcs,
                              form_compiler_parameters=form_compiler_parameters,
                              mat_type=mat_type,
                              sub_mat_type=sub_mat_type,
+                             diagonal=diagonal,
                              assembly_type="residual")