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): type {type} - """.format(**self.infos) + """.format( + **self.infos + ) table = 'FeatureValue\n' features = self.features lines = [] for f in features.keys(): - ind = features[f] - if ind: - lines.append("{feature}{tick}".format(feature=f, tick="")) - else: - lines.append("{feature}{tick}".format(feature=f, tick="")) + ind = features[f] + if ind: + lines.append( + '{feature}{tick}'.format( + feature=f, tick='' + ) + ) + else: + lines.append( + '{feature}{tick}'.format( + feature=f, tick='' + ) + ) table += str.join("\n", lines) table = "{}
".format(table) - - return table_infos + table - + return table_infos + table