Apply 1 suggestion(s) to 1 file(s)

Co-authored-by: Frederik Hennig <frederik.hennig@fau.de>

src/pystencils/backend/reduction_op_mapping.py

@@ -12,27 +12,20 @@ _available_operator_interface: set[ReductionOp] = {
 
 
 def reduction_op_to_expr(op: ReductionOp, op1, op2) -> PsExpression:
-    if op in _available_operator_interface:
-        match op:
-            case ReductionOp.Add:
-                return PsAdd(op1, op2)
-            case ReductionOp.Sub:
-                return PsSub(op1, op2)
-            case ReductionOp.Mul:
-                return PsMul(op1, op2)
-            case ReductionOp.Div:
-                return PsDiv(op1, op2)
-            case _:
-                raise FreezeError(
-                    f"Found unsupported operation type for reduction assignments: {op}."
-                )
-    else:
-        match op:
-            case ReductionOp.Min:
-                return PsCall(PsMathFunction(MathFunctions.Min), [op1, op2])
-            case ReductionOp.Max:
-                return PsCall(PsMathFunction(MathFunctions.Max), [op1, op2])
-            case _:
-                raise FreezeError(
-                    f"Found unsupported operation type for reduction assignments: {op}."
-                )
+    match op:
+        case ReductionOp.Add:
+            return PsAdd(op1, op2)
+        case ReductionOp.Sub:
+            return PsSub(op1, op2)
+        case ReductionOp.Mul:
+            return PsMul(op1, op2)
+        case ReductionOp.Div:
+            return PsDiv(op1, op2)
+        case ReductionOp.Min:
+            return PsCall(PsMathFunction(MathFunctions.Min), [op1, op2])
+        case ReductionOp.Max:
+            return PsCall(PsMathFunction(MathFunctions.Max), [op1, op2])
+        case _:
+            raise FreezeError(
+                f"Found unsupported operation type for reduction assignments: {op}."
+            )