diff --git a/mmpretrain/datasets/gqa_dataset.py b/mmpretrain/datasets/gqa_dataset.py new file mode 100644 index 00000000..ef101fec --- /dev/null +++ b/mmpretrain/datasets/gqa_dataset.py @@ -0,0 +1 @@ +# Copyright (c) OpenMMLab. All rights reserved. diff --git a/tests/test_evaluation/test_metrics/test_scienceqa.py b/tests/test_evaluation/test_metrics/test_scienceqa.py index 7e97d0e7..5df50aa3 100644 --- a/tests/test_evaluation/test_metrics/test_scienceqa.py +++ b/tests/test_evaluation/test_metrics/test_scienceqa.py @@ -1,5 +1,4 @@ # Copyright (c) OpenMMLab. All rights reserved. -import torch from mmengine.evaluator import Evaluator from mmpretrain.structures import DataSample @@ -10,12 +9,12 @@ class TestScienceQAMetric: def test_evaluate(self): meta_info = { 'choices': ['A', 'B', 'C', 'D'], - 'prediction': 'A', + 'pred_answer': 'A', 'grade': 'grade1', 'subject': 'language science', - 'answer': 1, + 'gt_answer': 1, 'hint': 'hint', - 'image': torch.ones((3, 224, 224)) + 'has_image': True } data_sample = DataSample(metainfo=meta_info) data_samples = [data_sample for _ in range(10)]