Fix Keras nightly build script.

PiperOrigin-RevId: 519188628
applib-sg-1009 · Mar 24, 2023 · b661c9f · b661c9f
1 parent b6080ff
commit b661c9f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pip_build.py b/pip_build.py
@@ -441,7 +441,9 @@ def test_wheel(wheel_path, expected_version, requirements_path):
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument(
-        "--nightly", default=False, help="Whether this is for keras-nightly"
+        "--nightly",
+        action=argparse.BooleanOptionalAction,
+        help="Whether this is for the `keras-nightly` package.",
     )
     args = parser.parse_args()
     is_nightly = args.nightly