[Fix]: Fix UT

pull/1577/head
liuyuan 2023-05-22 11:55:08 +08:00
parent b0ad99afb9
commit 13e4d6c512
2 changed files with 4 additions and 4 deletions

View File

@ -0,0 +1 @@
# Copyright (c) OpenMMLab. All rights reserved.

View File

@ -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)]