Test versions validations (#1598)

* test * basedir
pytorch · Nov 16, 2023 · e4f5f0d · e4f5f0d
1 parent 86169a9
commit e4f5f0d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -17,7 +17,7 @@
 stable_version = os.getenv("MATRIX_STABLE_VERSION")
 package_type = os.getenv("MATRIX_PACKAGE_TYPE")
 target_os = os.getenv("TARGET_OS")
-BASE_DIR =  Path(__file__).parent.parent
+BASE_DIR =  Path(__file__).parent.parent.parent
 
 is_cuda_system = gpu_arch_type == "cuda"
 NIGHTLY_ALLOWED_DELTA = 3
@@ -85,6 +85,8 @@ def check_version(package: str) -> None:
                     raise RuntimeError(
                         f"{module['name']} version mismatch, expected {release_version[module['name']]} for channel {channel}. But its {module_version}"
                     )
+                else:
+                     print(f"{module['name']} version actual: {module_version} expected: {release_version[module['name']]} for channel {channel}.")
 
     else:
         print(f"Skip version check for channel {channel} as stable version is None")