import argparse


def main() -> None:
    from mlx.core import __version__

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--version",
        action="version",
        version=__version__,
        help="Print the version number.",
    )
    parser.add_argument(
        "--cmake-dir",
        action="store_true",
        help="Print the path to the MLX CMake module directory.",
    )
    args = parser.parse_args()
    if args.cmake_dir:
        from pathlib import Path

        print(Path(__file__).parent)


if __name__ == "__main__":
    main()