Pass bazel options to requirements_update and requirements_nightly_up…

…date commands
jax-ml · Aug 16, 2024 · 12e8bf4 · 12e8bf4
1 parent 527a4b8
commit 12e8bf4
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/build/build.py b/build/build.py
@@ -622,20 +622,17 @@ def main():
       python_version=python_version,
   )
 
-  if args.requirements_update:
+  if args.requirements_update or args.requirements_nightly_update:
+    if args.requirements_update:
+      task = "//build:requirements.update"
+    else:  # args.requirements_nightly_update
+      task = "//build:requirements_nightly.update"
     update_command = ([bazel_path] + args.bazel_startup_options +
-      ["run", "--verbose_failures=true", "//build:requirements.update"])
+      ["run", "--verbose_failures=true", task, *args.bazel_options])
     print(" ".join(update_command))
     shell(update_command)
     return
 
-  if args.requirements_nightly_update:
-    update_nightly_command = ([bazel_path] + args.bazel_startup_options +
-      ["run", "--verbose_failures=true", "//build:requirements_nightly.update"])
-    print(" ".join(update_nightly_command))
-    shell(update_nightly_command)
-    return
-
   if args.configure_only:
     return