From 3462257c2380a0dc9d32b080a32373ee8d05acb1 Mon Sep 17 00:00:00 2001 From: mcDandy Date: Sat, 20 Jun 2026 15:14:34 +0200 Subject: [PATCH] Fix of a fix --- more_math/Parser/UnifiedMathVisitor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/more_math/Parser/UnifiedMathVisitor.py b/more_math/Parser/UnifiedMathVisitor.py index 72db0a8..90f36ae 100644 --- a/more_math/Parser/UnifiedMathVisitor.py +++ b/more_math/Parser/UnifiedMathVisitor.py @@ -2957,9 +2957,9 @@ class UnifiedMathVisitor(MathExprVisitor): shape_val = yield ctx.indexExpr() if self._is_list(shape_val): - shape = self._to_int(shape_val, ctx, "tensor") + shape = [self._to_int(v, ctx, "tensor") for v in shape_val] elif self._is_tensor(shape_val): - shape = self._to_int(shape_val, ctx, "tensor").flatten().tolist() + shape = [self._to_int(v, ctx, "tensor") for v in shape_val.flatten().tolist()] else: shape = [self._to_int(shape_val, ctx, "tensor")]