Skip to content

Commit

Permalink
fixed typo in get_video
Browse files Browse the repository at this point in the history
  • Loading branch information
pauladkisson committed Sep 25, 2023
1 parent 36f9394 commit ac6a7c0
Showing 1 changed file with 2 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -550,8 +550,8 @@ def get_video(self, start_frame=None, end_frame=None) -> np.ndarray:
(j - start_frame_in_cycle) * self._num_channels
+ (i + bool(start_left_in_cycle)) * self._num_raw_per_cycle
)
for j in range(end_left_in_cycle): # Add remaining frames from last (incomplete) cycle
index.append(j - start_frame_in_cycle) * self._num_channels + num_cycles * self._num_raw_per_cycle
for j in range(end_left_in_cycle): # Add remaining frames from last (incomplete) cycle)
index.append((j - start_frame_in_cycle) * self._num_channels + num_cycles * self._num_raw_per_cycle)
video = raw_video[index]
return video

Expand Down

0 comments on commit ac6a7c0

Please sign in to comment.