From dcd1941b94c27ae2b20bafed6348395e15cf3efc Mon Sep 17 00:00:00 2001 From: Francis Lata Date: Tue, 28 Jan 2025 03:06:43 -0800 Subject: [PATCH] fix validation test case --- test/external/external_test_datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/external/external_test_datasets.py b/test/external/external_test_datasets.py index a07e4e7d92..5e8bc1533f 100644 --- a/test/external/external_test_datasets.py +++ b/test/external/external_test_datasets.py @@ -157,7 +157,7 @@ class TestOpenImagesDataset(ExternalTestDatasets): ref_dataloader = self._create_ref_dataloader(base_dir, ann_file, "val") transform = GeneralizedRCNNTransform(img_size, img_mean, img_std) - for ((tinygrad_img, _), (ref_img, _)) in zip(tinygrad_dataloader, ref_dataloader): + for ((tinygrad_img, _, _), (ref_img, _)) in zip(tinygrad_dataloader, ref_dataloader): ref_img, _ = transform(ref_img.unsqueeze(0)) np.testing.assert_equal(tinygrad_img.numpy(), ref_img.tensors.transpose(1, 3).numpy())