Skip to content

Commit 4d9575a

Browse files
osalpekarosalpekar
committed
[Nova] More comprehensive Smoke Tests for Torchvision
1 parent edb3a80 commit 4d9575a

File tree

1 file changed

+149
-3
lines changed

1 file changed

+149
-3
lines changed

‎test/smoke_test.py

Lines changed: 149 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,150 @@
1-
import torch
1+
"""Run smoke tests"""
2+
23
import torchvision
3-
import torchvision.datasets as dset
4-
import torchvision.transforms
4+
import torchvision._internally_replaced_utils # noqa: F401
5+
import torchvision._utils # noqa: F401
6+
import torchvision.datasets # noqa: F401
7+
import torchvision.datasets._optical_flow # noqa: F401
8+
import torchvision.datasets._stereo_matching # noqa: F401
9+
import torchvision.datasets.caltech # noqa: F401
10+
import torchvision.datasets.celeba # noqa: F401
11+
import torchvision.datasets.cifar # noqa: F401
12+
import torchvision.datasets.cityscapes # noqa: F401
13+
import torchvision.datasets.clevr # noqa: F401
14+
import torchvision.datasets.coco # noqa: F401
15+
import torchvision.datasets.country211 # noqa: F401
16+
import torchvision.datasets.dtd # noqa: F401
17+
import torchvision.datasets.eurosat # noqa: F401
18+
import torchvision.datasets.fakedata # noqa: F401
19+
import torchvision.datasets.fer2013 # noqa: F401
20+
import torchvision.datasets.fgvc_aircraft # noqa: F401
21+
import torchvision.datasets.flickr # noqa: F401
22+
import torchvision.datasets.flowers102 # noqa: F401
23+
import torchvision.datasets.folder # noqa: F401
24+
import torchvision.datasets.food101 # noqa: F401
25+
import torchvision.datasets.gtsrb # noqa: F401
26+
import torchvision.datasets.hmdb51 # noqa: F401
27+
import torchvision.datasets.imagenet # noqa: F401
28+
import torchvision.datasets.inaturalist # noqa: F401
29+
import torchvision.datasets.kinetics # noqa: F401
30+
import torchvision.datasets.kitti # noqa: F401
31+
import torchvision.datasets.lfw # noqa: F401
32+
import torchvision.datasets.lsun # noqa: F401
33+
import torchvision.datasets.mnist # noqa: F401
34+
import torchvision.datasets.omniglot # noqa: F401
35+
import torchvision.datasets.oxford_iiit_pet # noqa: F401
36+
import torchvision.datasets.pcam # noqa: F401
37+
import torchvision.datasets.phototour # noqa: F401
38+
import torchvision.datasets.places365 # noqa: F401
39+
import torchvision.datasets.rendered_sst2 # noqa: F401
40+
import torchvision.datasets.sbd # noqa: F401
41+
import torchvision.datasets.sbu # noqa: F401
42+
import torchvision.datasets.semeion # noqa: F401
43+
import torchvision.datasets.stanford_cars # noqa: F401
44+
import torchvision.datasets.stl10 # noqa: F401
45+
import torchvision.datasets.sun397 # noqa: F401
46+
import torchvision.datasets.svhn # noqa: F401
47+
import torchvision.datasets.ucf101 # noqa: F401
48+
import torchvision.datasets.usps # noqa: F401
49+
import torchvision.datasets.utils # noqa: F401
50+
import torchvision.datasets.video_utils # noqa: F401
51+
import torchvision.datasets.vision # noqa: F401
52+
import torchvision.datasets.voc # noqa: F401
53+
import torchvision.datasets.widerface # noqa: F401
54+
import torchvision.extension # noqa: F401
55+
import torchvision.io # noqa: F401
56+
import torchvision.io._load_gpu_decoder # noqa: F401
57+
import torchvision.io._video_opt # noqa: F401
58+
import torchvision.io.image # noqa: F401
59+
import torchvision.io.video # noqa: F401
60+
import torchvision.io.video_reader # noqa: F401
61+
import torchvision.models # noqa: F401
62+
import torchvision.models._api # noqa: F401
63+
import torchvision.models._meta # noqa: F401
64+
import torchvision.models._utils # noqa: F401
65+
import torchvision.models.alexnet # noqa: F401
66+
import torchvision.models.convnext # noqa: F401
67+
import torchvision.models.densenet # noqa: F401
68+
import torchvision.models.detection # noqa: F401
69+
import torchvision.models.detection._utils # noqa: F401
70+
import torchvision.models.detection.anchor_utils # noqa: F401
71+
import torchvision.models.detection.backbone_utils # noqa: F401
72+
import torchvision.models.detection.faster_rcnn # noqa: F401
73+
import torchvision.models.detection.fcos # noqa: F401
74+
import torchvision.models.detection.generalized_rcnn # noqa: F401
75+
import torchvision.models.detection.image_list # noqa: F401
76+
import torchvision.models.detection.keypoint_rcnn # noqa: F401
77+
import torchvision.models.detection.mask_rcnn # noqa: F401
78+
import torchvision.models.detection.retinanet # noqa: F401
79+
import torchvision.models.detection.roi_heads # noqa: F401
80+
import torchvision.models.detection.rpn # noqa: F401
81+
import torchvision.models.detection.ssd # noqa: F401
82+
import torchvision.models.detection.ssdlite # noqa: F401
83+
import torchvision.models.detection.transform # noqa: F401
84+
import torchvision.models.efficientnet # noqa: F401
85+
import torchvision.models.googlenet # noqa: F401
86+
import torchvision.models.inception # noqa: F401
87+
import torchvision.models.maxvit # noqa: F401
88+
import torchvision.models.mnasnet # noqa: F401
89+
import torchvision.models.mobilenet # noqa: F401
90+
import torchvision.models.mobilenetv2 # noqa: F401
91+
import torchvision.models.mobilenetv3 # noqa: F401
92+
import torchvision.models.optical_flow # noqa: F401
93+
import torchvision.models.optical_flow._utils # noqa: F401
94+
import torchvision.models.optical_flow.raft # noqa: F401
95+
import torchvision.models.quantization # noqa: F401
96+
import torchvision.models.quantization.googlenet # noqa: F401
97+
import torchvision.models.quantization.inception # noqa: F401
98+
import torchvision.models.quantization.mobilenet # noqa: F401
99+
import torchvision.models.quantization.mobilenetv2 # noqa: F401
100+
import torchvision.models.quantization.mobilenetv3 # noqa: F401
101+
import torchvision.models.quantization.resnet # noqa: F401
102+
import torchvision.models.quantization.shufflenetv2 # noqa: F401
103+
import torchvision.models.quantization.utils # noqa: F401
104+
import torchvision.models.regnet # noqa: F401
105+
import torchvision.models.resnet # noqa: F401
106+
import torchvision.models.segmentation # noqa: F401
107+
import torchvision.models.segmentation._utils # noqa: F401
108+
import torchvision.models.segmentation.deeplabv3 # noqa: F401
109+
import torchvision.models.segmentation.fcn # noqa: F401
110+
import torchvision.models.segmentation.lraspp # noqa: F401
111+
import torchvision.models.shufflenetv2 # noqa: F401
112+
import torchvision.models.squeezenet # noqa: F401
113+
import torchvision.models.swin_transformer # noqa: F401
114+
import torchvision.models.vgg # noqa: F401
115+
import torchvision.models.video # noqa: F401
116+
import torchvision.models.video.mvit # noqa: F401
117+
import torchvision.models.video.resnet # noqa: F401
118+
import torchvision.models.video.s3d # noqa: F401
119+
import torchvision.models.vision_transformer # noqa: F401
120+
import torchvision.ops # noqa: F401
121+
import torchvision.ops._box_convert # noqa: F401
122+
import torchvision.ops._register_onnx_ops # noqa: F401
123+
import torchvision.ops._utils # noqa: F401
124+
import torchvision.ops.boxes # noqa: F401
125+
import torchvision.ops.ciou_loss # noqa: F401
126+
import torchvision.ops.deform_conv # noqa: F401
127+
import torchvision.ops.diou_loss # noqa: F401
128+
import torchvision.ops.drop_block # noqa: F401
129+
import torchvision.ops.feature_pyramid_network # noqa: F401
130+
import torchvision.ops.focal_loss # noqa: F401
131+
import torchvision.ops.giou_loss # noqa: F401
132+
import torchvision.ops.misc # noqa: F401
133+
import torchvision.ops.poolers # noqa: F401
134+
import torchvision.ops.ps_roi_align # noqa: F401
135+
import torchvision.ops.ps_roi_pool # noqa: F401
136+
import torchvision.ops.roi_align # noqa: F401
137+
import torchvision.ops.roi_pool # noqa: F401
138+
import torchvision.ops.stochastic_depth # noqa: F401
139+
import torchvision.transforms # noqa: F401
140+
import torchvision.transforms._pil_constants # noqa: F401
141+
import torchvision.transforms._presets # noqa: F401
142+
import torchvision.transforms.autoaugment # noqa: F401
143+
import torchvision.transforms.functional # noqa: F401
144+
import torchvision.transforms.functional_pil # noqa: F401
145+
import torchvision.transforms.functional_tensor # noqa: F401
146+
import torchvision.transforms.transforms # noqa: F401
147+
import torchvision.utils # noqa: F401
148+
import torchvision.version # noqa: F401
149+
150+
print("torchvision version is ", torchvision.__version__)

0 commit comments

Comments
 (0)