Skip to content

Commit f1b474d

Browse files
committed
Renaming methods.
1 parent 5828cfd commit f1b474d

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

torchvision/csrc/io/video_reader/video_reader.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -564,7 +564,7 @@ torch::List<torch::Tensor> probeVideo(
564564

565565
} // namespace
566566

567-
torch::List<torch::Tensor> readVideoFromMemory(
567+
torch::List<torch::Tensor> read_video_from_memory(
568568
torch::Tensor input_video,
569569
double seekFrameMargin,
570570
int64_t getPtsOnly,
@@ -608,7 +608,7 @@ torch::List<torch::Tensor> readVideoFromMemory(
608608
audioTimeBaseDen);
609609
}
610610

611-
torch::List<torch::Tensor> readVideoFromFile(
611+
torch::List<torch::Tensor> read_video_from_file(
612612
std::string videoPath,
613613
double seekFrameMargin,
614614
int64_t getPtsOnly,
@@ -653,20 +653,20 @@ torch::List<torch::Tensor> readVideoFromFile(
653653
audioTimeBaseDen);
654654
}
655655

656-
torch::List<torch::Tensor> probeVideoFromMemory(torch::Tensor input_video) {
656+
torch::List<torch::Tensor> probe_video_from_memory(torch::Tensor input_video) {
657657
return probeVideo(false, input_video, "");
658658
}
659659

660-
torch::List<torch::Tensor> probeVideoFromFile(std::string videoPath) {
660+
torch::List<torch::Tensor> probe_video_from_file(std::string videoPath) {
661661
torch::Tensor dummy_input_video = torch::ones({0});
662662
return probeVideo(true, dummy_input_video, videoPath);
663663
}
664664

665665
TORCH_LIBRARY_FRAGMENT(video_reader, m) {
666-
m.def("read_video_from_memory", readVideoFromMemory);
667-
m.def("read_video_from_file", readVideoFromFile);
668-
m.def("probe_video_from_memory", probeVideoFromMemory);
669-
m.def("probe_video_from_file", probeVideoFromFile);
666+
m.def("read_video_from_memory", read_video_from_memory);
667+
m.def("read_video_from_file", read_video_from_file);
668+
m.def("probe_video_from_memory", probe_video_from_memory);
669+
m.def("probe_video_from_file", probe_video_from_file);
670670
}
671671

672672
} // namespace video_reader

0 commit comments

Comments
 (0)