diff --git a/dolark/model.py b/dolark/model.py index 743342a..f2d620c 100644 --- a/dolark/model.py +++ b/dolark/model.py @@ -467,17 +467,17 @@ def __str__(self): from dolo.misc.termcolor import colored from numpy import zeros - s = "Model:\n------\nname: \"{name}\"\ntype: \"{type}\"\n".format(**self.infos) + s = 'Model:\n------\nname: "{name}"\ntype: "{type}"\n'.format(**self.infos) s += "\nFeatures:\n----------\n" features = self.features for f in features.keys(): - ind = features[f] - if ind: - s += "{feature} {tick}".format(feature=f, tick=colored("✓", "green")) - else: - s += "{feature} {tick}".format(feature=f, tick=colored("✗", "red")) - s += "\n" + ind = features[f] + if ind: + s += "{feature} {tick}".format(feature=f, tick=colored("✓", "green")) + else: + s += "{feature} {tick}".format(feature=f, tick=colored("✗", "red")) + s += "\n" return s @@ -498,20 +498,28 @@ def _repr_html_(self):