Simplify [v[0], ..., v[k]] -> v for tensor valued v

FEniCS · Jan 4, 2025 · 766da29 · 766da29
1 parent 5df0e45
commit 766da29
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ufl/tensors.py b/ufl/tensors.py
@@ -61,7 +61,7 @@ def __new__(cls, *expressions):
         # Simplify [v[0], v[1], ..., v[k]] -> v
         if (
             all(isinstance(e, Indexed) for e in expressions)
-            and e0.ufl_operands[0].ufl_shape == (len(expressions),)
+            and e0.ufl_operands[0].ufl_shape[0] == len(expressions)
             and all(e.ufl_operands[0] == e0.ufl_operands[0] for e in expressions)
         ):
             indices = list(chain.from_iterable(e.ufl_operands[1].indices() for e in expressions))