fix 2D/3D vector recovery...

firedrakeproject · Jul 25, 2023 · d50e646 · d50e646
1 parent 8dac6c9
commit d50e646
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gusto/recovery/recovery.py b/gusto/recovery/recovery.py
@@ -332,7 +332,7 @@ def find_eff_coords(V0):
     vec_DG1 = VectorFunctionSpace(mesh, DG1_element)
     x = SpatialCoordinate(mesh)
 
-    if isinstance(V0.ufl_element(), VectorElement):
+    if isinstance(V0.ufl_element(), VectorElement) or V0.ufl_element().value_size() > 1:
         eff_coords_list = []
         V0_coords_list = []