Update IREE python tool name
Match the package name change from https://github.com/openxla/iree/commit/5b60156
Also fix the pylint timeout warning
Change-Id: I6ea22e2b5ea3b61b37761ee135407def9d5d6415
diff --git a/download_iree_compiler.py b/download_iree_compiler.py
index 17ced92..8419f74 100755
--- a/download_iree_compiler.py
+++ b/download_iree_compiler.py
@@ -119,7 +119,8 @@
snapshot = None
if args.tag_name:
r = requests.get((f"{args.release_url}/tags/{args.tag_name}"),
- auth=('user', 'pass'))
+ auth=('user', 'pass'),
+ timeout=60)
if r.status_code != 200:
print(
f"!!!!!IREE snapshot can't be found with tag {args.tag_name}, "
@@ -127,7 +128,7 @@
sys.exit(1)
snapshot = r.json()
else:
- r = requests.get(args.release_url, auth=('user', 'pass'))
+ r = requests.get(args.release_url, auth=('user', 'pass'), timeout=60)
if r.status_code != 200:
print("Not getting the right snapshot information. Status code: %d",
r.status_code)
@@ -156,8 +157,7 @@
tmp_dir = Path(os.getenv("OUT")) / "tmp"
whl_file = download_artifact(snapshot["assets"],
- ["iree_tools_tflite", "linux", "x86_64.whl"],
- tmp_dir)
+ ["iree_tools_tflite", ".whl"], tmp_dir)
tar_file = download_artifact(
snapshot["assets"], ["linux-x86_64.tar"], tmp_dir)
