diff --git a/src/formatter.py b/src/formatter.py index a8b69fa..932062f 100644 --- a/src/formatter.py +++ b/src/formatter.py @@ -37,7 +37,7 @@ class OutNoStatFormatter(Formatter): self.representation = "" def header_repeat(self): - return ["Lemma", "Representative_form", "RF_scenario"] + return ["Lemma", "Representative_form", "RF_msd", "RF_scenario"] def header_right(self): return ["Joint_representative_form", "Frequency"] @@ -50,10 +50,10 @@ class OutNoStatFormatter(Formatter): rep = representations[idx] if rep is None: self.representation += " " + word.lemma - return [word.lemma, word.lemma, "lemma_fallback"] + return [word.lemma, word.lemma, "", "lemma_fallback"] else: self.representation += " " + rep - return [word.lemma, rep, "ok"] + return [word.lemma, rep, word.msd, "ok"] def content_right(self, freq): rep = re.sub(' +', ' ', self.representation.strip())