From 013aff2f2e417380b9f012cf6a3ea53a8f90d7be Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Mon, 2 Sep 2019 11:15:28 +0200
Subject: [PATCH] Fixup: lint
---
pystencils/astnodes.py | 4 ++--
pystencils/backends/cbackend.py | 17 ++++++-----------
2 files changed, 8 insertions(+), 13 deletions(-)
diff --git a/pystencils/astnodes.py b/pystencils/astnodes.py
index 55617313..81127f66 100644
--- a/pystencils/astnodes.py
+++ b/pystencils/astnodes.py
@@ -3,7 +3,7 @@ from typing import Any, List, Optional, Sequence, Set, Union
import sympy as sp
-from pystencils.data_types import TypedSymbol, cast_func, create_type, TypedImaginaryUnit
+from pystencils.data_types import TypedImaginaryUnit, TypedSymbol, cast_func, create_type
from pystencils.field import Field
from pystencils.kernelparameters import FieldPointerSymbol, FieldShapeSymbol, FieldStrideSymbol
from pystencils.sympyextensions import fast_subs
@@ -537,7 +537,7 @@ class SympyAssignment(Node):
loop_counters.add(LoopOverCoordinate.get_loop_counter_symbol(i))
result.update(loop_counters)
result.update(self._lhs_symbol.atoms(sp.Symbol))
- result = { r for r in result if not isinstance(r, TypedImaginaryUnit)}
+ result = {r for r in result if not isinstance(r, TypedImaginaryUnit)}
return result
@property
diff --git a/pystencils/backends/cbackend.py b/pystencils/backends/cbackend.py
index e38b8fde..f2f478c6 100644
--- a/pystencils/backends/cbackend.py
+++ b/pystencils/backends/cbackend.py
@@ -212,9 +212,7 @@ class CBackend:
method_name = "_print_" + cls.__name__
if hasattr(self, method_name):
return getattr(self, method_name)(node)
- raise NotImplementedError(self.__class__.__name__ +
- " does not support node of type " +
- node.__class__.__name__)
+ raise NotImplementedError(self.__class__.__name__ + " does not support node of type " + node.__class__.__name__)
def _print_Type(self, node):
return str(node)
@@ -428,7 +426,7 @@ class CustomSympyPrinter(CCodePrinter):
elif expr.func in infix_functions:
return "(%s %s %s)" % (self._print(
expr.args[0]), infix_functions[expr.func],
- self._print(expr.args[1]))
+ self._print(expr.args[1]))
elif expr.func == int_power_of_2:
return "(1 << (%s))" % (self._print(expr.args[0]))
elif expr.func == int_div:
@@ -444,7 +442,7 @@ class CustomSympyPrinter(CCodePrinter):
elif dtype.numpy_dtype == np.float64:
return res + '.0' if '.' not in res else res
elif dtype.numpy_dtype == np.complex64:
- return f"{self._typed_number(number.real, np.float32)} + {self._typed_number(number.real, np.float32).replace('f', 'if')}"
+ return f"{self._typed_number(number.real, np.float32)} + {self._typed_number(number.real, np.float32).replace('f', 'if')}" # noqa
elif dtype.numpy_dtype == np.complex128:
return f"{self._typed_number(number.real, np.float64)} + {self._typed_number(number.real, np.float64)}i"
else:
@@ -469,8 +467,7 @@ class CustomSympyPrinter(CCodePrinter):
end=self._print(end),
expr=self._print(expr.function),
increment=str(1),
- condition=self._print(var) + ' <= ' +
- self._print(end) # if start < end else '>='
+ condition=self._print(var) + ' <= ' + self._print(end) # if start < end else '>='
)
return code
@@ -493,8 +490,7 @@ class CustomSympyPrinter(CCodePrinter):
end=self._print(end),
expr=self._print(expr.function),
increment=str(1),
- condition=self._print(var) + ' <= ' +
- self._print(end) # if start < end else '>='
+ condition=self._print(var) + ' <= ' + self._print(end) # if start < end else '>='
)
return code
@@ -566,8 +562,7 @@ class VectorizedCustomSympyPrinter(CustomSympyPrinter):
return self.instruction_set['rsqrt'].format(
self._print(expr.args[0]))
else:
- return "({})".format(self._print(1 /
- sp.sqrt(expr.args[0])))
+ return "({})".format(self._print(1 / sp.sqrt(expr.args[0])))
elif isinstance(expr, vec_any):
expr_type = get_type_of_expression(expr.args[0])
if type(expr_type) is not VectorType:
--
GitLab