Files
mlflow--mlflow/dev/validate_release_version.py
2026-07-13 13:22:34 +08:00

26 lines
580 B
Python

import argparse
from packaging.version import Version
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser()
parser.add_argument(
"--version", help="Release version to validate, e.g., '1.2.3'", required=True
)
return parser.parse_args()
def main() -> None:
args = parse_args()
version = Version(args.version)
msg = (
f"Invalid release version: '{args.version}', "
"must be in the format of <major>.<minor>.<micro>"
)
assert len(version.release) == 3, msg
if __name__ == "__main__":
main()