test_datasets_video_utils_opt.py 356 Bytes
Newer Older
1
import unittest
limm's avatar
limm committed
2

3
import test_datasets_video_utils
limm's avatar
limm committed
4
from torchvision import set_video_backend  # noqa: 401
5

6
7
# Disabling the video backend switching temporarily
# set_video_backend('video_reader')
8
9


limm's avatar
limm committed
10
if __name__ == "__main__":
11
12
    suite = unittest.TestLoader().loadTestsFromModule(test_datasets_video_utils)
    unittest.TextTestRunner(verbosity=1).run(suite)