[fbsync] Fix failing smoke test (#7388)

Reviewed By: vmoens

Differential Revision: D44416545

fbshipit-source-id: bf700b79ef207eed7f6714ffb38e437e35dcc086

Author: NicolasHug

test/smoke_test.py

@@ -1,8 +1,8 @@
 """Run smoke tests"""
 
 import os
+import sys
 from pathlib import Path
-from sys import platform
 
 import torch
 import torch.nn as nn
@@ -37,7 +37,7 @@ def smoke_test_compile() -> None:
         out = model(x)
         print(f"torch.compile model output: {out.shape}")
     except RuntimeError:
-        if platform == "win32":
+        if sys.platform == "win32":
             print("Successfully caught torch.compile RuntimeError on win")
         elif sys.version_info >= (3, 11, 0):
             print("Successfully caught torch.compile RuntimeError on Python 3.11")