Commit 07a3f195 authored by Martin Bauer's avatar Martin Bauer
Browse files

lbmpy notebook cleanup Part2

parent d3aa7c68
......@@ -113,7 +113,7 @@ class SimplificationStrategy(object):
def print_assignment_collection(title, c):
text = title
if self.restrict_symbols:
text += "\n".join([str(e) for e in c.get(self.restrict_symbols)])
text += "\n".join([str(e) for e in c.new_filtered(self.restrict_symbols).main_assignments])
else:
text += (" " * 3 + (" " * 3).join(str(c).splitlines(True)))
return text
......@@ -129,7 +129,8 @@ class SimplificationStrategy(object):
def print_assignment_collection(title, c):
text = '<h5 style="padding-bottom:10px">%s</h5> <div style="padding-left:20px;">' % (title, )
if self.restrict_symbols:
text += "\n".join(["$$" + sp.latex(e) + '$$' for e in c.get(self.restrict_symbols)])
text += "\n".join(["$$" + sp.latex(e) + '$$'
for e in c.new_filtered(self.restrict_symbols).main_assignments])
else:
# noinspection PyProtectedMember
text += c._repr_html_()
......
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