Modify dictionary to function on cellnames

firedrakeproject · Nov 29, 2024 · bb0754b · bb0754b
1 parent 2be9fb2
commit bb0754b
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/firedrake/mesh.py b/firedrake/mesh.py
@@ -66,10 +66,10 @@
 }
 
 
-_supported_embedded_cell_types_and_gdims = [(ufl.Cell('interval'), 2),
-                                            (ufl.Cell('triangle'), 3),
-                                            (ufl.Cell("quadrilateral"), 3),
-                                            (ufl.TensorProductCell(ufl.Cell('interval'), ufl.Cell('interval')), 3)]
+_supported_embedded_cell_types_and_gdims = [('interval', 2),
+                                            ('triangle', 3),
+                                            ("quadrilateral", 3),
+                                            ("interval * interval", 3)]
 
 
 unmarked = -1
@@ -2765,7 +2765,7 @@ def init_cell_orientations(self, expr):
         import firedrake.function as function
         import firedrake.functionspace as functionspace
 
-        if (self.ufl_cell(), self.geometric_dimension()) not in _supported_embedded_cell_types_and_gdims:
+        if (self.ufl_cell().cellname(), self.geometric_dimension()) not in _supported_embedded_cell_types_and_gdims:
             raise NotImplementedError('Only implemented for intervals embedded in 2d and triangles and quadrilaterals embedded in 3d')
 
         if hasattr(self, '_cell_orientations'):