Merge pull request jax-ml#7523 from yashk2810:bazel

PiperOrigin-RevId: 389013729
Shuhua886 · Aug 5, 2021 · d83450d · d83450d
2 parents be2f9b9 + b24574f
commit d83450d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build/build.py b/build/build.py
@@ -212,7 +212,7 @@ def check_bazel_version(bazel_path):
     version_output = shell([bazel_path, "--version"])
   except subprocess.CalledProcessError:
     return False
-  match = re.search("Build label: *([0-9\\.]+)[^0-9\\.]", version_output)
+  match = re.search(r"bazel *([0-9\\.]+)", version_output)
   if match is None:
     return False
   actual_ints = [int(x) for x in match.group(1).split(".")]