From e80267ae0610ae8543af68bed742995209d818eb Mon Sep 17 00:00:00 2001
From: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Date: Thu, 31 Mar 2022 23:25:50 +0800
Subject: [PATCH] [Fix] Fix unit tests of evaluator (#162)
---
tests/test_evaluator/test_base_evaluator.py | 5 ++---
1 file changed, 2 insertions(+), 3 deletions(-)
diff --git a/tests/test_evaluator/test_base_evaluator.py b/tests/test_evaluator/test_base_evaluator.py
index bed31b1f..042d2fb8 100644
--- a/tests/test_evaluator/test_base_evaluator.py
+++ b/tests/test_evaluator/test_base_evaluator.py
@@ -79,10 +79,9 @@ def generate_test_results(size, batch_size, pred, label):
bs_residual = size % batch_size
for i in range(num_batch):
bs = bs_residual if i == num_batch - 1 else batch_size
- data_batch = [(np.zeros(
- (3, 10, 10)), BaseDataElement(data={'label': label}))
+ data_batch = [(np.zeros((3, 10, 10)), BaseDataElement(label=label))
for _ in range(bs)]
- predictions = [BaseDataElement(data={'pred': pred}) for _ in range(bs)]
+ predictions = [BaseDataElement(pred=pred) for _ in range(bs)]
yield (data_batch, predictions)
--
GitLab