From 96b5cbf286a29882b496faee3b9fe3be481d8bb3 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Tue, 28 Jan 2025 12:44:21 +0100
Subject: [PATCH] Fix lint

---
 src/pystencils/backend/kernelcreation/freeze.py | 5 ++---
 src/pystencils/codegen/driver.py                | 2 +-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py
index f5f207acf..1238f16af 100644
--- a/src/pystencils/backend/kernelcreation/freeze.py
+++ b/src/pystencils/backend/kernelcreation/freeze.py
@@ -1,9 +1,8 @@
 from typing import overload, cast, Any
 from functools import reduce
-from operator import add, mul, sub, truediv
+from operator import add, mul, sub
 
 import sympy as sp
-import sympy.core.relational
 import sympy.logic.boolalg
 from sympy.codegen.ast import AssignmentBase, AugmentedAssignment
 
@@ -184,7 +183,7 @@ class FreezeExpressions:
         assert isinstance(lhs, PsSymbolExpr)
 
         orig_lhs_symb = lhs.symbol
-        dtype = rhs.dtype # TODO: kernel with (implicit) up/downcasts?
+        dtype = rhs.dtype  # TODO: kernel with (implicit) up/downcasts?
 
         # replace original symbol with pointer-based type used for export
         orig_lhs_symb_as_ptr = PsSymbol(orig_lhs_symb.name, PsPointerType(dtype))
diff --git a/src/pystencils/codegen/driver.py b/src/pystencils/codegen/driver.py
index 8b8ecd15b..b47ad8a9e 100644
--- a/src/pystencils/codegen/driver.py
+++ b/src/pystencils/codegen/driver.py
@@ -7,7 +7,7 @@ from .config import CreateKernelConfig, OpenMpConfig, VectorizationConfig, AUTO
 from .kernel import Kernel, GpuKernel, GpuThreadsRange
 from .properties import PsSymbolProperty, FieldShape, FieldStride, FieldBasePtr, ReductionPointerVariable
 from .parameters import Parameter
-from ..backend.ast.expressions import PsSymbolExpr, PsMemAcc, PsConstantExpr, PsExpression
+from ..backend.ast.expressions import PsSymbolExpr, PsMemAcc, PsConstantExpr
 from ..sympyextensions.binop_mapping import binop_str_to_expr
 
 from ..types import create_numeric_type, PsIntegerType, PsScalarType
-- 
GitLab