From 97f8f9bf4723526b9a9063129df93e5e2345796a Mon Sep 17 00:00:00 2001 From: Jeremy Reizenstein Date: Fri, 27 Jan 2023 03:56:53 -0800 Subject: [PATCH] test fix Reviewed By: shapovalov Differential Revision: D42780711 fbshipit-source-id: 075fcae5097147b782f7ffc935f5430b824f58fd --- tests/implicitron/test_batch_sampler.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/implicitron/test_batch_sampler.py b/tests/implicitron/test_batch_sampler.py index aba57551..9f3732a7 100644 --- a/tests/implicitron/test_batch_sampler.py +++ b/tests/implicitron/test_batch_sampler.py @@ -48,7 +48,8 @@ class MockDataset(DatasetBase): for i in idx: self.frame_annots[i]["frame_annotation"].sequence_name = seq_name - def get_frame_numbers_and_timestamps(self, idxs): + def get_frame_numbers_and_timestamps(self, idxs, subset_filter=None): + assert subset_filter is None out = [] for idx in idxs: frame_annotation = self.frame_annots[idx]["frame_annotation"]