Skip to content

Commit 13bd5c5

Browse files
shalgifacebook-github-bot
authored andcommitted
[TorchCodec] fix typo
Summary: decorder_args -> decoder_args Reviewed By: ahmadsharif1 Differential Revision: D60463535 fbshipit-source-id: 081db62b108edc339426421db3546a1ec812e89f
1 parent e43a20f commit 13bd5c5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/torchcodec/_samplers/video_clip_sampler.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ class VideoClipSampler(nn.Module):
115115
Args:
116116
video_args (`VideoArgs`): The video args
117117
sampler_args (`SamplerArgs`): The sampler args. Can be TimeBasedSamplerArgs or IndexBasedSamplerArgs
118-
decorder_args (`DecoderArgs`): Decoder args contain value needs for decoder, for example, thread count
118+
decoder_args (`DecoderArgs`): Decoder args contain value needs for decoder, for example, thread count
119119
120120
Example:
121121
>>> video_args = VideoArgs(desired_width=224, desired_height=224)
@@ -130,12 +130,12 @@ def __init__(
130130
self,
131131
video_args: VideoArgs,
132132
sampler_args: SamplerArgs,
133-
decorder_args: Union[None, DecoderArgs] = None,
133+
decoder_args: Union[None, DecoderArgs] = None,
134134
) -> None:
135135
super().__init__()
136136
self.video_args = video_args
137137
self.sampler_args = sampler_args
138-
self.decorder_args = DecoderArgs() if decorder_args is None else decorder_args
138+
self.decoder_args = DecoderArgs() if decoder_args is None else decoder_args
139139

140140
def forward(self, video_data: Tensor) -> Union[List[Any]]:
141141
"""Sample video clips from the video data
@@ -162,7 +162,7 @@ def forward(self, video_data: Tensor) -> Union[List[Any]]:
162162
video_decoder,
163163
width=target_width,
164164
height=target_height,
165-
num_threads=self.decorder_args.num_threads,
165+
num_threads=self.decoder_args.num_threads,
166166
)
167167

168168
clips: List[Any] = []

0 commit comments

Comments
 (0)