Make numpy check optional (#149356)

atalman · pytorchbot · commit f0e7257ecbed · 2025-05-12T22:34:09.000Z
We may want to skip numpy smoke tests. Hence making it optional Pull Request resolved: #149356 Approved by: https://github.com/ZainRizvi (cherry picked from commit 6e2b266)
diff --git a/.ci/pytorch/smoke_test/smoke_test.py b/.ci/pytorch/smoke_test/smoke_test.py
@@ -76,10 +76,13 @@ def read_release_matrix():
 
 
 def test_numpy():
-    import numpy as np
+    try:
+        import numpy as np
 
-    x = np.arange(5)
-    torch.tensor(x)
+        x = np.arange(5)
+        torch.tensor(x)
+    except ImportError:
+        print("Numpy check skipped. Numpy is not installed.")
 
 
 def check_version(package: str) -> None:
@@ -410,6 +413,7 @@ def main() -> None:
     smoke_test_conv2d()
     test_linalg()
     test_numpy()
+
     if is_cuda_system:
         test_linalg("cuda")
         test_cuda_gds_errors_captured()
