Skip to content

Commit beb4bb7

Browse files
authored
Fix failing smoke test (#7388)
1 parent 5d9bc5e commit beb4bb7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/smoke_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,8 @@
11
"""Run smoke tests"""
22

33
import os
4+
import sys
45
from pathlib import Path
5-
from sys import platform
66

77
import torch
88
import torch.nn as nn
@@ -37,7 +37,7 @@ def smoke_test_compile() -> None:
3737
out = model(x)
3838
print(f"torch.compile model output: {out.shape}")
3939
except RuntimeError:
40-
if platform == "win32":
40+
if sys.platform == "win32":
4141
print("Successfully caught torch.compile RuntimeError on win")
4242
elif sys.version_info >= (3, 11, 0):
4343
print("Successfully caught torch.compile RuntimeError on Python 3.11")

0 commit comments

Comments
 (0)