Fix Keras nightly build script.

PiperOrigin-RevId: 519188628
2023-03-24 11:03:27 -07:00 · 2023-03-24 11:03:27 -07:00 · b661c9f73d
commit b661c9f73d
parent b6080ffbd3
1 changed files with 3 additions and 1 deletions
--- 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