From 16a5cd55d9b349a186ad62ad4ce574a7e4dba96c Mon Sep 17 00:00:00 2001 From: Matthew Brett Date: Fri, 10 Nov 2023 11:10:30 +0000 Subject: [PATCH] Set musllinux wheel dependency --- print_deps.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/print_deps.py b/print_deps.py index 83a7e81..6106637 100755 --- a/print_deps.py +++ b/print_deps.py @@ -2,6 +2,7 @@ """ Echo dependencies for given environment """ +import os from pathlib import Path from argparse import ArgumentParser, RawDescriptionHelpFormatter @@ -19,11 +20,13 @@ def get_build_requirements(repo_path): def get_numpy_requirement(py_ver): major, minor, *_ = py_ver.split('.') assert major == "3" - np_version = "1.22.2" + musl = os.environ.get('MB_ML_LIBC') == 'musllinux' + # musllinux wheels started at 1.25.0 + np_version = "1.22.2" if not musl else '1.25.0' minor = int(minor) if minor >= 12: np_version = "1.26.0" - elif minor >= 11: + elif minor >= 11 and not musl: np_version = "1.23.2" return np_version