test

Author: goanpeca

scripts/install_torch_xla.py

@@ -10,9 +10,9 @@ def main(python_version: str, xla_version: str) -> None:
     temp_dir = Path(tempfile.mkdtemp()) / "torch_xla"
     temp_dir.mkdir(parents=True, exist_ok=True)
     pyver = python_version.replace(".", "")
-    url = f"https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-{xla_version}-cp{pyver}-cp{pyver}-linux_x86_64.whl"
+    url = f"https://storage.googleapis.com/pytorch-xla-releases/wheels/tpuvm/torch_xla-{xla_version}"
     suffix = f"-cp{pyver}-cp{pyver}-linux_x86_64.whl"
-    r = urlopen(url)
+    r = urlopen(url + suffix)
     wheel_path = temp_dir / "torch_xla.zip"
 
     with open(wheel_path, "b+w") as f:
@@ -24,7 +24,7 @@ def main(python_version: str, xla_version: str) -> None:
     dist = [p for p in os.listdir(temp_dir) if "dist-info" in p]
     if dist:
         pkg_name = dist[0].split(".dist-info")[0]
-        wheel_path = wheel_path.rename(f"{pkg_name}{suffix}")
+        wheel_path = wheel_path.rename(temp_dir / f"{pkg_name}{suffix}")
         cmd = ["uv", "pip", "install", str(wheel_path)]
         os.system(" ".join(cmd))
 

tests/ignite/handlers/conftest.py

@@ -1,6 +1,5 @@
 import subprocess
 import time
-import traceback
 from pathlib import Path
 from unittest.mock import Mock
 
@@ -29,9 +28,7 @@ def visdom_server():
             time.sleep(1)
             vis = Visdom(server=vd_hostname, port=vd_port, raise_exceptions=True)
             break
-        except ConnectionError as e:
-            print(f"Visdom server not ready yet, retrying {ii + 1}/5", e)
-            traceback.print_exc()
+        except ConnectionError:
             continue
 
     assert vis and vis.check_connection()