Skip to content
Snippets Groups Projects
Commit e20ebf94 authored by Martin Bauer's avatar Martin Bauer
Browse files

PEP8 compliance

parent 4b34c96c
No related branches found
No related tags found
No related merge requests found
...@@ -90,10 +90,9 @@ class LbmWeightInfo(CustomCppCode): ...@@ -90,10 +90,9 @@ class LbmWeightInfo(CustomCppCode):
def __init__(self, lb_method): def __init__(self, lb_method):
weights = [str(w.evalf()) for w in lb_method.weights] weights = [str(w.evalf()) for w in lb_method.weights]
code = "const double __attribute__((unused)) %s [] = { %s };\n" % ( w_sym = LbmWeightInfo.WEIGHTS_SYMBOL
LbmWeightInfo.WEIGHTS_SYMBOL.name, ",".join(weights)) code = "const double __attribute__((unused)) %s [] = { %s };\n" % (w_sym.name, ",".join(weights))
super(LbmWeightInfo, self).__init__(code, symbols_read=set(), super(LbmWeightInfo, self).__init__(code, symbols_read=set(), symbols_defined={w_sym})
symbols_defined={LbmWeightInfo.WEIGHTS_SYMBOL})
def create_lattice_boltzmann_boundary_kernel(pdf_field, index_field, lb_method, boundary_functor, def create_lattice_boltzmann_boundary_kernel(pdf_field, index_field, lb_method, boundary_functor,
......
...@@ -83,6 +83,7 @@ def n_phases_correction_function(c, beta, power=2): ...@@ -83,6 +83,7 @@ def n_phases_correction_function(c, beta, power=2):
(-beta * (1 - c) ** power, c > 1), (-beta * (1 - c) ** power, c > 1),
(c ** 2 * (1 - c) ** 2, True)) (c ** 2 * (1 - c) ** 2, True))
def n_phases_correction_function_wrong(c, beta, power=2): def n_phases_correction_function_wrong(c, beta, power=2):
return sp.Piecewise((-beta * c ** power, c < 0), return sp.Piecewise((-beta * c ** power, c < 0),
(-beta * (1 - c) ** power, c > 1), (-beta * (1 - c) ** power, c > 1),
......
...@@ -58,7 +58,7 @@ def create_stream_pull_only_kernel(stencil, numpy_arr=None, src_field_name="src" ...@@ -58,7 +58,7 @@ def create_stream_pull_only_kernel(stencil, numpy_arr=None, src_field_name="src"
if numpy_arr is None: if numpy_arr is None:
src = Field.create_generic(src_field_name, dim, index_shape=(len(stencil),), src = Field.create_generic(src_field_name, dim, index_shape=(len(stencil),),
layout=generic_layout, dtype=generic_field_type) layout=generic_layout, dtype=generic_field_type)
dst = Field.create_generic(dst_field_name, dim, index_shape=(len(stencil),), dst = Field.create_generic(dst_field_name, dim, index_shape=(len(stencil),),
layout=generic_layout, dtype=generic_field_type) layout=generic_layout, dtype=generic_field_type)
else: else:
src = Field.create_from_numpy_array(src_field_name, numpy_arr, index_dimensions=1) src = Field.create_from_numpy_array(src_field_name, numpy_arr, index_dimensions=1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment