From b25c2c234f34d8d8ce5e8d5cd8f64d2efee6867d Mon Sep 17 00:00:00 2001 From: Rob Kirby Date: Wed, 13 Dec 2023 10:17:46 -0600 Subject: [PATCH] Fix what table gets imported --- FIAT/quadrature_schemes.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/FIAT/quadrature_schemes.py b/FIAT/quadrature_schemes.py index 1f1118f35..31297dd14 100644 --- a/FIAT/quadrature_schemes.py +++ b/FIAT/quadrature_schemes.py @@ -30,9 +30,12 @@ # NumPy import numpy +from FIAT.quadrature import (QuadratureRule, make_quadrature, + make_tensor_product_quadrature, map_quadrature) # FIAT -from FIAT.reference_element import QUADRILATERAL, HEXAHEDRON, TENSORPRODUCT, TRIANGLE, TETRAHEDRON, UFCTriangle, UFCTetrahedron, default_simplex -from FIAT.quadrature import QuadratureRule, make_quadrature, make_tensor_product_quadrature, map_quadrature +from FIAT.reference_element import (HEXAHEDRON, QUADRILATERAL, TENSORPRODUCT, + TETRAHEDRON, TRIANGLE, UFCTetrahedron, + UFCTriangle, default_simplex) def create_quadrature(ref_el, degree, scheme="default"): @@ -337,8 +340,10 @@ def xg_scheme(ref_el, degree): http://dx.doi.org/10.1016/j.camwa.2009.10.027 """ dim = ref_el.get_spatial_dimension() - if dim == 2 or dim == 3: + if dim == 2: from FIAT.xg_quad_data import triangle_table as table + elif dim == 3: + from FIAT.xg_quad_data import tetrahedron_table as table else: raise ValueError(f"Xiao-Gambutas rule not availale for {dim} dimensions.") try: