423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 03:42:50 +08:00
Code Issues Packages Projects Releases Wiki Activity

test fix

Browse Source 
Reviewed By: shapovalov

Differential Revision: D42780711

fbshipit-source-id: 075fcae5097147b782f7ffc935f5430b824f58fd
This commit is contained in:
Jeremy Reizenstein 2023-01-27 03:56:53 -08:00 committed by Facebook GitHub Bot
parent 7e750a3786
commit 97f8f9bf47
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/implicitron/test_batch_sampler.py
View File

@ -48,7 +48,8 @@ class MockDataset(DatasetBase):
for i in idx: for i in idx:
self.frame_annots[i]["frame_annotation"].sequence_name = seq_name 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 = [] out = []
for idx in idxs: for idx in idxs:
frame_annotation = self.frame_annots[idx]["frame_annotation"] frame_annotation = self.frame_annots[idx]["frame_annotation"]