From b661c9f73d8860cda6425e3c5b0b1ec145329dc1 Mon Sep 17 00:00:00 2001
From: Francois Chollet <fchollet@google.com>
Date: Fri, 24 Mar 2023 11:03:27 -0700
Subject: [PATCH] Fix Keras nightly build script.

PiperOrigin-RevId: 519188628
---
 pip_build.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/pip_build.py b/pip_build.py
index 1232892cd..6f4d53e8f 100644
--- 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