Add handling of zero (happens with extract_block)

FEniCS · Oct 1, 2024 · 886bc85 · 886bc85
1 parent 65766dc
commit 886bc85
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ufl/algorithms/apply_derivatives.py b/ufl/algorithms/apply_derivatives.py
@@ -1139,6 +1139,8 @@ def compute_gprimeterm(ngrads, vval, vcomp, wshape, wcomp):
                             gprimesum = gprimesum + compute_gprimeterm(
                                 ngrads, vval, vcomp, wshape, wcomp
                             )
+                elif isinstance(v, Zero):
+                    pass
 
                 else:
                     if wshape != ():