Fix lint

src/pystencils/backend/ast/vector.py

@@ -92,11 +92,9 @@ class PsVecHorizontal(PsBinOp, PsVectorOp):
     def structurally_equal(self, other: PsAstNode) -> bool:
         if not isinstance(other, PsVecHorizontal):
             return False
-        return (
-                super().structurally_equal(other)
+        return (super().structurally_equal(other)
                 and self._lanes == other._lanes
-                and self._reduction_op == other._reduction_op
-        )
+                and self._reduction_op == other._reduction_op)
 
 
 class PsVecMemAcc(PsExpression, PsLvalue, PsVectorOp):

src/pystencils/backend/platforms/cuda.py

@@ -89,7 +89,7 @@ class CudaPlatform(GenericGpu):
 
         if isinstance(dtype, PsScalarType) and func in (NumericLimitsFunctions.Min, NumericLimitsFunctions.Max):
             assert isinstance(dtype, PsIeeeFloatType)
-            defines = { NumericLimitsFunctions.Min: "NEG_INFINITY", NumericLimitsFunctions.Max: "POS_INFINITY" }
+            defines = {NumericLimitsFunctions.Min: "NEG_INFINITY", NumericLimitsFunctions.Max: "POS_INFINITY"}
 
             return PsLiteralExpr(PsLiteral(defines[func], dtype))
 
@@ -170,8 +170,8 @@ class CudaPlatform(GenericGpu):
                     case ReductionOp.Sub:
                         # workaround for unsupported atomicSub: use atomic add
                         # similar to OpenMP reductions: local copies (negative sign) are added at the end
-                        call.function = CFunction(f"atomicAdd", [ptr_expr.dtype, symbol_expr.dtype],
-                                          PsCustomType("void"))
+                        call.function = CFunction("atomicAdd", [ptr_expr.dtype, symbol_expr.dtype],
+                                                  PsCustomType("void"))
                         call.args = (ptr_expr, symbol_expr)
                     case _:
                         call.function = CFunction(f"atomic{op.name}", [ptr_expr.dtype, symbol_expr.dtype],

src/pystencils/backend/transformations/loop_vectorizer.py

@@ -240,14 +240,14 @@ class LoopVectorizer:
                 )
 
                 return PsBlock(
-                    simd_init_local_reduction_vars +
-                    [
+                    simd_init_local_reduction_vars
+                    + [
                         simd_stop_decl,
                         simd_step_decl,
                         simd_loop
-                    ] +
-                    simd_writeback_local_reduction_vars +
-                    [
+                    ]
+                    + simd_writeback_local_reduction_vars
+                    + [
                         trailing_start_decl,
                         trailing_loop,
                     ]
@@ -258,13 +258,13 @@ class LoopVectorizer:
 
             case LoopVectorizer.TrailingItersTreatment.NONE:
                 return PsBlock(
-                    simd_init_local_reduction_vars +
-                    [
+                    simd_init_local_reduction_vars
+                    + [
                         simd_stop_decl,
                         simd_step_decl,
                         simd_loop,
-                    ] +
-                    simd_writeback_local_reduction_vars
+                    ]
+                    + simd_writeback_local_reduction_vars
                 )
 
     @overload

src/pystencils/compound_op_mapping.py

-from operator import truediv, mul, sub, add
-
 from .backend.ast.expressions import PsExpression, PsCall, PsAdd, PsSub, PsMul, PsDiv
 from .backend.exceptions import FreezeError
 from .backend.functions import PsMathFunction, MathFunctions

src/pystencils/jit/gpu_cupy.py

@@ -11,7 +11,6 @@ except ImportError:
 from ..codegen import Target
 from ..field import FieldType
 
-from ..types import PsType, PsPointerType
 from .jit import JitBase, JitError, KernelWrapper
 from ..codegen import (
     Kernel,
@@ -19,7 +18,7 @@ from ..codegen import (
     Parameter,
 )
 from ..codegen.properties import FieldShape, FieldStride, FieldBasePtr
-from ..types import PsStructType, PsPointerType
+from ..types import PsType, PsStructType, PsPointerType
 
 from ..include import get_pystencils_include_path
 

src/pystencils/sympyextensions/reduction.py

@@ -19,14 +19,18 @@ class ReductionAssignment(AssignmentBase):
     Attributes:
     ===========
 
-    binop : CompoundOp
+    reduction_op : ReductionOp
        Enum for binary operation being applied in the assignment, such as "Add" for "+", "Sub" for "-", etc.
     """
-    reduction_op = None  # type: ReductionOp
+    _reduction_op = None  # type: ReductionOp
 
     @property
     def reduction_op(self):
-        return self.reduction_op
+        return self._reduction_op
+
+    @reduction_op.setter
+    def reduction_op(self, op):
+        self._reduction_op = op
 
 
 class AddReductionAssignment(ReductionAssignment):