Skip to content

Commit 641f6ca

Browse files
committed
Fix default arg and print statement bugs in XLA update script
1 parent bb43aa2 commit 641f6ca

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tools/update_xla_hash.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
def update_xla_hash(xla_commit, xla_repo, workspace_file_path, gh_token):
1919
# Verify that the workspace_file exists
2020
if not os.path.isfile(workspace_file_path):
21-
raise ValueError(f"Workspace file '{workspace_file}' does not exist")
21+
raise ValueError(f"Workspace file '{workspace_file_path}' does not exist")
2222

2323
# If we were given a GH auth token, use it to make sure that the commit
2424
# exists and convert a branch name to a commit hash
@@ -59,7 +59,7 @@ def update_xla_hash(xla_commit, xla_repo, workspace_file_path, gh_token):
5959
# Edit the commit hash, sha256 hash, and repo
6060
contents = re.sub(
6161
'XLA_COMMIT = "[a-z0-9]*"',
62-
f'XLA_COMIT = "{xla_commit_hash}"',
62+
f'XLA_COMMIT = "{xla_commit_hash}"',
6363
contents,
6464
flags=re.M,
6565
)
@@ -100,7 +100,7 @@ def parse_args():
100100
)
101101
arg_parser.add_argument(
102102
"--workspace-file",
103-
default=".jax_rocm_plugin/third_party/xla/workspace.bzl",
103+
default="jax_rocm_plugin/third_party/xla/workspace.bzl",
104104
help="Path to the workspace.bzl file to put the hash. Defaults to ./third_party/xla/workspace.bzl.",
105105
)
106106
arg_parser.add_argument(

0 commit comments

Comments
 (0)