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

PEP8 compliance

parent 4b34c96c
Branches
Tags
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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment