From 6be6c92e5d8792b1c91d3bf7e07242521c831ae2 Mon Sep 17 00:00:00 2001 From: James Betker Date: Fri, 6 Nov 2020 20:38:34 -0700 Subject: [PATCH] Fix yet ANOTHER OBO error in multi_frame_dataset --- codes/data/multi_frame_dataset.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/codes/data/multi_frame_dataset.py b/codes/data/multi_frame_dataset.py index c8f14fe1..c82355c9 100644 --- a/codes/data/multi_frame_dataset.py +++ b/codes/data/multi_frame_dataset.py @@ -19,10 +19,9 @@ class MultiFrameDataset(BaseUnsupervisedImageDataset): frames_needed = self.num_frames search_idx = chunk_index while frames_needed > 0: - if source_name in self.chunk_name(search_idx): - frames_needed -= 1 - search_idx -= 1 - else: + frames_needed -= 1 + search_idx -= 1 + if source_name not in self.chunk_name(search_idx): search_idx += 1 break