Commit 80f81c97 authored by Markus Holzer's avatar Markus Holzer
Browse files

Fix minor issues

parent 0ef8c3d5
......@@ -357,13 +357,13 @@ class CenteredCumulantBasedLbMethod(AbstractLbMethod):
cqe = conserved_quantity_equations
relaxation_info_dict = dict()
subexpressions_relaxation_rates = []
if symbolic_relaxation_rates:
subexpressions_relaxation_rates, sd = self._generate_symbolic_relaxation_matrix()
for i, cumulant in enumerate(cumulant_to_relaxation_info_dict):
relaxation_info_dict[cumulant] = RelaxationInfo(cumulant_to_relaxation_info_dict[cumulant][0],
sd[i, i])
else:
subexpressions_relaxation_rates = []
relaxation_info_dict = cumulant_to_relaxation_info_dict
if cqe is None:
......
......@@ -251,13 +251,13 @@ class CentralMomentBasedLbMethod(AbstractLbMethod):
cqe = conserved_quantity_equations
relaxation_info_dict = dict()
subexpressions_relaxation_rates = []
if symbolic_relaxation_rates:
subexpressions_relaxation_rates, sd = self._generate_symbolic_relaxation_matrix()
for i, cumulant in enumerate(cumulant_to_relaxation_info_dict):
relaxation_info_dict[cumulant] = RelaxationInfo(cumulant_to_relaxation_info_dict[cumulant][0],
for i, moment in enumerate(moment_to_relaxation_info_dict):
relaxation_info_dict[moment] = RelaxationInfo(moment_to_relaxation_info_dict[moment][0],
sd[i, i])
else:
subexpressions_relaxation_rates = []
relaxation_info_dict = moment_to_relaxation_info_dict
if cqe is None:
......@@ -299,7 +299,8 @@ class CentralMomentBasedLbMethod(AbstractLbMethod):
# 4) Now, put it all together.
all_acs = [] if pdfs_to_c_transform.absorbs_conserved_quantity_equations else [cqe]
all_acs += [pdfs_to_c_eqs, collision_eqs]
subexpressions_relaxation_rates = AssignmentCollection(subexpressions_relaxation_rates)
all_acs += [subexpressions_relaxation_rates, pdfs_to_c_eqs, collision_eqs]
subexpressions = [ac.all_assignments for ac in all_acs]
subexpressions += c_post_to_pdfs_eqs.subexpressions
main_assignments = c_post_to_pdfs_eqs.main_assignments
......
Markdown is supported
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