Skip to content

Commit 0e62c34

Browse files
authored
fix code format (#7377)
1 parent caf12f8 commit 0e62c34

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/smoke_test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,13 +27,15 @@ def smoke_test_torchvision_read_decode() -> None:
2727
if img_png.ndim != 3 or img_png.numel() < 100:
2828
raise RuntimeError(f"Unexpected shape of img_png: {img_png.shape}")
2929

30+
3031
def smoke_test_compile() -> None:
3132
model = resnet50().cuda()
3233
model = torch.compile(model)
3334
x = torch.randn(1, 3, 224, 224, device="cuda")
3435
out = model(x)
3536
print(f"torch.compile model output: {out.shape}")
3637

38+
3739
def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
3840
img = read_image(str(SCRIPT_DIR / ".." / "gallery" / "assets" / "dog2.jpg")).to(device)
3941

@@ -73,6 +75,5 @@ def main() -> None:
7375
smoke_test_torchvision_resnet50_classify("mps")
7476

7577

76-
7778
if __name__ == "__main__":
7879
main()

0 commit comments

Comments
 (0)