From 96df6b6a533b0fe52a3f82576764fc6533429e07 Mon Sep 17 00:00:00 2001 From: Daniel Martinek Date: Fri, 13 Mar 2026 15:26:14 +0100 Subject: [PATCH] Allways stack when converting color representations --- more_math/Parser/UnifiedMathVisitor.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/more_math/Parser/UnifiedMathVisitor.py b/more_math/Parser/UnifiedMathVisitor.py index 6b62dfd..cedbdb9 100644 --- a/more_math/Parser/UnifiedMathVisitor.py +++ b/more_math/Parser/UnifiedMathVisitor.py @@ -3284,11 +3284,7 @@ class UnifiedMathVisitor(MathExprVisitor): # Value v = max_rgb - # Return format - if num_args <= 2: - return torch.stack([h, s, v], dim=-1) - else: - return [h, s, v] + return torch.stack([h, s, v], dim=-1) def visitHsvToRgbFunc(self, ctx): num_args = len(ctx.expr()) @@ -3371,11 +3367,8 @@ class UnifiedMathVisitor(MathExprVisitor): g = g + m b = b + m - # Return format - if num_args <= 2: - return torch.stack([r, g, b], dim=-1) - else: - return [r, g, b] + return torch.stack([r, g, b], dim=-1) + def visitEntropyFunc(self, ctx): val = self._promote_to_tensor((yield ctx.expr()))