Formatting

Author: scotts

src/torchcodec/decoders/_core/VideoDecoder.cpp

@@ -560,8 +560,8 @@ std::vector<int64_t> VideoDecoder::getKeyFrameIndices(int streamIndex) {
   std::vector<int64_t> keyFrameIndices;
   const StreamInfo& streamInfo = streamInfos_[streamIndex];
   for (const FrameInfo& frameInfo : streamInfo.keyFrames) {
-    keyFrameIndices.push_back(
-        getKeyFrameIndexForPtsUsingScannedIndex(streamInfo.keyFrames, frameInfo.pts));
+    keyFrameIndices.push_back(getKeyFrameIndexForPtsUsingScannedIndex(
+        streamInfo.keyFrames, frameInfo.pts));
   }
 
   return keyFrameIndices;

src/torchcodec/decoders/_core/video_decoder_ops.py

@@ -257,7 +257,9 @@ def get_frames_by_pts_in_range_abstract(
 
 
 @register_fake("torchcodec_ns::get_key_frame_indices")
-def get_key_frame_indices_abstract(decoder: torch.Tensor, *, stream_index: int) -> List[int]:
+def get_key_frame_indices_abstract(
+    decoder: torch.Tensor, *, stream_index: int
+) -> List[int]:
     return []