Skip to content
Snippets Groups Projects
Unverified Commit e80267ae authored by Zaida Zhou's avatar Zaida Zhou Committed by GitHub
Browse files

[Fix] Fix unit tests of evaluator (#162)

parent 1927bc77
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment