Skip to content
Snippets Groups Projects

[FIX] PSM generation

Merged Markus Holzer requested to merge holzer/lbmpy:fixPSM into master
Compare and
2 files
+ 71
2
Preferences
Compare changes
Files
2
@@ -124,7 +124,8 @@ class AbstractLbMethod(abc.ABC):
substitutions = [Assignment(e[1], e[0]) for e in subexpressions.items()]
if relaxation_rates_modifier is not None:
symbolic_relaxation_rates = [r * relaxation_rates_modifier for r in symbolic_relaxation_rates]
for srr in symbolic_relaxation_rates:
assert isinstance(srr, sp.Symbol)
else:
for srr in symbolic_relaxation_rates:
assert isinstance(srr, sp.Symbol)
return substitutions, sp.diag(*symbolic_relaxation_rates)