Skip to content

Commit 09ada7c

Browse files
committed
Fix return type
1 parent 1670dc6 commit 09ada7c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/torchcodec/decoders/_core/video_decoder_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -259,8 +259,8 @@ def get_frames_by_pts_in_range_abstract(
259259
@register_fake("torchcodec_ns::_get_key_frame_indices")
260260
def get_key_frame_indices_abstract(
261261
decoder: torch.Tensor, *, stream_index: int
262-
) -> List[int]:
263-
return []
262+
) -> torch.Tensor:
263+
return torch.empty([], dtype=torch.int)
264264

265265

266266
@register_fake("torchcodec_ns::get_json_metadata")

0 commit comments

Comments
 (0)