From 9434be8d8754bbf743a55237e3c4cbdf29449dd4 Mon Sep 17 00:00:00 2001 From: Miklos Homolya Date: Mon, 8 Feb 2016 16:48:32 +0000 Subject: [PATCH] fix ufc_cell --- FIAT/reference_element.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/FIAT/reference_element.py b/FIAT/reference_element.py index ff7dcc372..3dd07d377 100644 --- a/FIAT/reference_element.py +++ b/FIAT/reference_element.py @@ -716,9 +716,9 @@ def ufc_cell( cell ): else: celltype = cell.cellname() - if celltype == "TensorProductCell": - # cell is a UFL cell - return TensorProductCell(ufc_cell(cell._A), ufc_cell(cell._B)) + if " * " in celltype: + # Tensor product cell + return TensorProductCell(*map(ufc_cell, celltype.split(" * "))) elif celltype == "quadrilateral": return FiredrakeQuadrilateral() elif celltype == "interval":