Skip to content

Commit

Permalink
Update frame and clip transform functions
Browse files Browse the repository at this point in the history
  • Loading branch information
SohamTilekar committed Mar 22, 2024
1 parent 2a64f40 commit cf9b8da
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions vidiopy/video/VideoFileClip.py
Original file line number Diff line number Diff line change
Expand Up @@ -113,7 +113,9 @@ def __eq__(self, other) -> bool:
#################

@requires_start_end
def fl_frame_transform(self, func: Callable, *args, **kwargs) -> Self:
def fl_frame_transform(
self, func: Callable[..., npt.NDArray], *args, **kwargs
) -> Self:
"""
Applies a function to each frame of the video clip.
Expand Down Expand Up @@ -150,7 +152,9 @@ def fl_frame_transform(self, func: Callable, *args, **kwargs) -> Self:
return self

@requires_fps
def fl_clip_transform(self, func, *args, **kwargs) -> Self:
def fl_clip_transform(
self, func: Callable[..., npt.NDArray], *args, **kwargs
) -> Self:
"""
Applies a function to each frame of the video clip along with its timestamp.
Expand Down

0 comments on commit cf9b8da

Please sign in to comment.