yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-09-16 19:53:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve Label and MultiLabel __str__ and __repr__ (#2202)

Browse Source 
* Add proper annotation to MultiLabel and slightly improve __str__ for MultiLabel and Label

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
This commit is contained in:
Sara Zan 2022-02-17 16:37:02 +01:00 committed by GitHub
parent 542d7b8128
commit c93b365f98
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
haystack/schema.py
View File

@ -481,11 +481,20 @@ class Label:
return str(self.to_dict())
def __str__(self):
return str(self.to_dict())
return f"<Label: {self.to_dict()}>"
@dataclass
class MultiLabel:
labels: List[Label]
query: str
answers: List[str]
no_answer: bool
document_ids: List[str]
document_contents: List[str]
gold_offsets_in_contexts: List[Dict]
gold_offsets_in_documents: List[Dict]
def __init__(self, labels: List[Label], drop_negative_labels=False, drop_no_answers=False):
"""
There are often multiple `Labels` associated with a single query. For example, there can be multiple annotated
@ -499,7 +508,6 @@ class MultiLabel:
:param drop_negative_labels: Whether to drop negative labels from that group (e.g. thumbs down feedback from UI)
:param drop_no_answers: Whether to drop labels that specify the answer is impossible
"""
# drop duplicate labels and remove negative labels if needed.
labels = list(set(labels))
if drop_negative_labels:
@ -588,7 +596,7 @@ class MultiLabel:
return str(self.to_dict())
def __str__(self):
return str(self.to_dict())
return f"<MultiLabel: {self.to_dict()}>"
def _pydantic_dataclass_from_dict(dict: dict, pydantic_dataclass_type) -> Any: