From bb410e0f1b10274ae751641e03e1847d6402f759 Mon Sep 17 00:00:00 2001 From: Patrick Farrell Date: Fri, 13 Mar 2020 14:05:17 +0000 Subject: [PATCH] Allow VectorElement to take a variant argument. --- ufl/finiteelement/mixedelement.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ufl/finiteelement/mixedelement.py b/ufl/finiteelement/mixedelement.py index f2a00c5a7..7c37543be 100644 --- a/ufl/finiteelement/mixedelement.py +++ b/ufl/finiteelement/mixedelement.py @@ -245,7 +245,7 @@ class VectorElement(MixedElement): "A special case of a mixed finite element where all elements are equal." def __init__(self, family, cell=None, degree=None, dim=None, - form_degree=None, quad_scheme=None): + form_degree=None, quad_scheme=None, variant=None): """ Create vector element (repeated mixed element) @@ -263,6 +263,8 @@ def __init__(self, family, cell=None, degree=None, dim=None, viewed as k-form), ignored if family is a FiniteElement quad_scheme The quadrature scheme (optional), ignored if family is a FiniteElement + variant + Hint for the local basis function variant (optional) """ if isinstance(family, FiniteElementBase): @@ -274,7 +276,8 @@ def __init__(self, family, cell=None, degree=None, dim=None, # Create sub element sub_element = FiniteElement(family, cell, degree, form_degree=form_degree, - quad_scheme=quad_scheme) + quad_scheme=quad_scheme, + variant=variant) # Set default size if not specified if dim is None: