Skip to content
Snippets Groups Projects

Rework of the Cumulant-Based Method

Merged Frederik Hennig requested to merge da15siwa/lbmpy:cumulant_method_rework into master
Viewing commit dd9178d5
Show latest version
2 files
+ 6
5
Preferences
Compare changes
Files
2
@@ -9,7 +9,7 @@ from itertools import product
from lbmpy.methods.abstractlbmethod import LbmCollisionRule
from pystencils import Assignment, AssignmentCollection
from pystencils.stencil import inverse_direction
from pystencils.simp.subexpression_insertion import insert_subexpressions
from pystencils.simp.subexpression_insertion import insert_subexpressions, is_constant
from pystencils.sympyextensions import extract_most_common_factor, replace_second_order_products, subs_additive
from collections import defaultdict
@@ -328,7 +328,7 @@ def insert_pure_products(ac, symbols, **kwargs):
if isinstance(rhs, sp.Symbol) and rhs in symbols:
return True
elif isinstance(rhs, sp.Mul):
if all(arg in symbols for arg in rhs.args):
if all((is_constant(arg) or (arg in symbols)) for arg in rhs.args):
return True
return False