superset not tabulate_polynomial_set

FEniCS · Jan 8, 2024 · f299094 · f299094
1 parent d8d60f9
commit f299094
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/basix/polynomials.py b/python/basix/polynomials.py
@@ -166,7 +166,7 @@ def restriction(
     Returns:
         The restricted polyset type
     """
-    return _restriction(ptype.value, cell.value, restriction_cell.value)
+    return getattr(PolysetType, _restriction(ptype.value, cell.value, restriction_cell.value).name)
 
 
 def superset(cell: CellType, type1: PolysetType, type2: PolysetType) -> PolysetType:
@@ -212,4 +212,4 @@ def tabulate_polynomial_set(
     Returns:
         Tabulated polynomial set
     """
-    return getattr(PolysetType, _tps(celltype.value, ptype.value, degree, nderiv, pts).name)
+    return _tps(celltype.value, ptype.value, degree, nderiv, pts)