From 6e3f3d2c57dcd04d3b005ddf8a4c8e7a03c2eed9 Mon Sep 17 00:00:00 2001 From: Nishant Yadav Date: Tue, 24 Sep 2024 16:49:56 -0700 Subject: [PATCH] minor fix --- .../visual_question_answering/vqa_as_kvp_dataset.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/vision_datasets/visual_question_answering/vqa_as_kvp_dataset.py b/vision_datasets/visual_question_answering/vqa_as_kvp_dataset.py index 0bc90b5..d863f1e 100644 --- a/vision_datasets/visual_question_answering/vqa_as_kvp_dataset.py +++ b/vision_datasets/visual_question_answering/vqa_as_kvp_dataset.py @@ -34,9 +34,9 @@ class VQAAsKeyValuePairDataset(VisionDataset): def __init__(self, vqa_dataset: VisionDataset): """ - Initializes an instance of the ClassificationAsKeyValuePairDataset class. + Initializes an instance of the VQAAsKeyValuePairDataset class. Args: - classification_dataset (VisionDataset): The classification dataset to convert to key-value pair dataset. + vqa_dataset (VisionDataset): The VQA dataset to convert to key-value pair dataset. """ if vqa_dataset is None or vqa_dataset.dataset_info.type is not DatasetTypes.VISUAL_QUESTION_ANSWERING: @@ -77,7 +77,7 @@ def construct_schema(self) -> Dict[str, Any]: def construct_kvp_label_data(self, label: Dict[str, str]) -> Dict[str, Dict[str, str]]: """ - Convert the classification dataset label_name to the desired format for KVP annotation as defined by the SCHEMA_BASE. + Convert the VQA dataset label to the desired format for KVP annotation as defined by the SCHEMA_BASE. E.g. {"fields": {"answer": {"value": "yes"}}, {"rationale": {"value": "I see a dog."}}