make cache key from weakref.proxy

firedrakeproject · Dec 12, 2024 · 63b667b · 63b667b
1 parent 8db60cb
commit 63b667b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tsfc/fem.py b/tsfc/fem.py
@@ -8,6 +8,7 @@
 import gem
 import numpy
 import ufl
+import weakref
 from FIAT.orientation_utils import Orientation as FIATOrientation
 from FIAT.reference_element import UFCHexahedron, UFCSimplex, make_affine_mapping
 from FIAT.reference_element import TensorProductCell
@@ -298,7 +299,7 @@ def weight_expr(self):
         return self.quadrature_rule.weight_expression
 
     def make_basis_evaluation_key(self, finat_element, mt, entity_id):
-        domain = extract_unique_domain(mt.terminal)
+        domain = weakref.proxy(extract_unique_domain(mt.terminal))
         restriction = mt.restriction
         return (self, finat_element, mt.local_derivatives, domain, restriction, entity_id)