Skip to content

Commit

Permalink
Set musllinux wheel dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
matthew-brett committed Nov 10, 2023
1 parent 54eb854 commit 16a5cd5
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions print_deps.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
""" Echo dependencies for given environment
"""

import os
from pathlib import Path
from argparse import ArgumentParser, RawDescriptionHelpFormatter

Expand All @@ -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

Expand Down

0 comments on commit 16a5cd5

Please sign in to comment.