Merge branch 'JHopeCollins/cofunction_fieldsplit' of https://github.c…

…om/firedrakeproject/firedrake into JHopeCollins/cofunction_fieldsplit
firedrakeproject · Dec 16, 2024 · 8bf436a · 8bf436a
2 parents 026c2f0 + 38bf268
commit 8bf436a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/firedrake/formmanipulation.py b/firedrake/formmanipulation.py
@@ -154,7 +154,7 @@ def cofunction(self, o):
         # on a two-form then we return a zero form,
         # analogously to the off components of arguments.
         if len(self.blocks) == 2:
-            itest, itrial = self.blocks[0], self.blocks[1]
+            itest, itrial = self.blocks
             on_diag = (itest == itrial)
         else:
             on_diag = True
@@ -174,7 +174,7 @@ def cofunction(self, o):
                 from firedrake import MixedFunctionSpace
                 from pyop2 import MixedDat
                 W = MixedFunctionSpace([V_is[i] for i in indices])
-                c = Cofunction(W.dual(), val=MixedDat(o.subfunctions[i].dat
+                c = Cofunction(W.dual(), val=MixedDat(o.dat[i]
                                                       for i in indices))
         else:
             c = ZeroBaseForm(o.arguments())