juancopi81's picture
Duplicate from Whispering-GPT/whisper-youtube-2-hf_dataset
7288748
raw
history blame
No virus
795 Bytes
from typing import List
from video import YoutubeVideo
from transforming.transform import Transform
from utils import accepts_types
class BatchTransformer:
"""Class that applies multiple transforms to YouTube video object."""
def __init__(self, transforms: List[Transform]) -> None:
self._transforms = transforms
@property
def transforms(self) -> List[Transform]:
return self._transforms
@transforms.setter
def transforms(self, transforms: List[Transform]) -> None:
self._transforms = transforms
@accepts_types(list)
def apply(self, videos: List[YoutubeVideo]) -> List[YoutubeVideo]:
for transform in self._transforms:
videos = list(map(transform.apply, videos))
return videos