diff --git a/seqscore/validation.py b/seqscore/validation.py index d5ce16d..56c984b 100644 --- a/seqscore/validation.py +++ b/seqscore/validation.py @@ -26,6 +26,9 @@ class ValidationResult: repaired_labels: Optional[Tuple[str, ...]] = attrib( converter=tuplify_strs, default=() ) + tokens: Optional[Tuple[str, ...]] = attrib(converter=tuplify_strs, default=None) + labels: Optional[Tuple[str, ...]] = attrib(converter=tuplify_strs, default=None) + line_nums: Optional[Tuple[int, ...]] = attrib(default=None) def is_valid(self) -> bool: return not self.errors @@ -130,6 +133,10 @@ def validate_labels( if errors and repair: repaired_labels = encoding.repair_labels(labels, repair) - return ValidationResult(errors, len(labels), repaired_labels) + return ValidationResult(errors, len(labels), repaired_labels, + tokens=tokens, labels=labels, line_nums=line_nums + ) else: - return ValidationResult(errors, len(labels)) + return ValidationResult(errors, len(labels), + tokens=tokens, labels=labels, line_nums=line_nums + )