function space: fix collapse

firedrakeproject · Nov 12, 2024 · 49a4ff1 · 49a4ff1
1 parent 29c486d
commit 49a4ff1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/firedrake/functionspaceimpl.py b/firedrake/functionspaceimpl.py
@@ -843,8 +843,7 @@ def local_to_global_map(self, bcs, lgmap=None):
         return PETSc.LGMap().create(indices, bsize=bsize, comm=lgmap.comm)
 
     def collapse(self):
-        from firedrake import FunctionSpace
-        return FunctionSpace(self.mesh(), self.ufl_element())
+        return type(self)(self.mesh(), self.ufl_element())
 
 
 class RestrictedFunctionSpace(FunctionSpace):
@@ -1161,8 +1160,7 @@ def _ises(self):
         return self.dof_dset.field_ises
 
     def collapse(self):
-        from firedrake import MixedFunctionSpace
-        return MixedFunctionSpace([V_ for V_ in self])
+        return type(self)([V_ for V_ in self], self.mesh())
 
 
 class ProxyFunctionSpace(FunctionSpace):