fix cwd

pytorch · Jun 14, 2023 · 76e4188 · 76e4188
1 parent 068f545
commit 76e4188
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -18,7 +18,6 @@
 package_type = os.getenv("MATRIX_PACKAGE_TYPE")
 
 is_cuda_system = gpu_arch_type == "cuda"
-SCRIPT_DIR = Path(__file__).parent
 NIGHTLY_ALLOWED_DELTA = 3
 
 MODULES = [
@@ -206,10 +205,11 @@ def foo(x: torch.Tensor) -> torch.Tensor:
     x_pt2 = torch.compile(model, mode="max-autotune")(x)
 
 def smoke_test_modules():
+    cwd = os.getcwd()
     for module in MODULES:
         if module["repo"]:
-            if not os.path.exists(f"{SCRIPT_DIR}/{module['repo_name']}"):
-                print(f"Path does not exist: {SCRIPT_DIR}/{module['repo_name']}")
+            if not os.path.exists(f"{cwd}/{module['repo_name']}"):
+                print(f"Path does not exist: {cwd}/{module['repo_name']}")
                 subprocess.check_output(f"git clone --depth 1 {module['repo']}", stderr=subprocess.STDOUT, shell=True)
             try:
                 output = subprocess.check_output(