test fix

Reviewed By: shapovalov Differential Revision: D42780711 fbshipit-source-id: 075fcae5097147b782f7ffc935f5430b824f58fd

test fix
Reviewed By: shapovalov Differential Revision: D42780711 fbshipit-source-id: 075fcae5097147b782f7ffc935f5430b824f58fd
97f8f9bf · Jeremy Reizenstein · Facebook GitHub Bot · 7e750a37 · 97f8f9bf
Commit 97f8f9bf authored Jan 27, 2023 by Jeremy Reizenstein Committed by Facebook GitHub Bot Jan 27, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tests/implicitron/test_batch_sampler.py tests/implicitron/test_batch_sampler.py +2 -1

No files found.
--- 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"]