Refactor Numpy version getting

Author: matthew-brett

.github/workflows/wheels-linux.yml

@@ -56,7 +56,7 @@ jobs:
         with:
           python-version: ${{ matrix.python }}
       - name: Set Numpy version
-        run: echo "BUILD_DEPENDS=$(./get_numpy_version.py)" >> $GITHUB_ENV
+        run: echo "BUILD_DEPENDS=$(./get_numpy_version.py ${{ matrix.python }})" >> $GITHUB_ENV
       - name: Build Wheel
         run: .github/workflows/build.sh
       - uses: actions/upload-artifact@v3

.github/workflows/wheels-windows.yml

@@ -39,7 +39,7 @@ jobs:
           python-version: ${{ matrix.python }}
       - name: Set Numpy version
         run: |
-          $np_dep = python .\get_numpy_version.py
+          $np_dep = python .\get_numpy_version.py ${{ matrix.python }}
           echo "BUILD_DEPENDS=$np_dep" | Out-File -FilePath $Env:GITHUB_ENV -Encoding utf8 -Append
       - name: Build Wheel
         run: .github/workflows/build.ps1

get_numpy_version.py

@@ -3,14 +3,15 @@
 """
 
 import os
-import sys
 
 from argparse import ArgumentParser, RawDescriptionHelpFormatter
 
 
 def get_parser():
     parser = ArgumentParser(description=__doc__,  # Usage from docstring
                             formatter_class=RawDescriptionHelpFormatter)
+    parser.add_argument("py_ver",
+                        help='Python version e.g. 3.11')
     parser.add_argument('-t', '--type', default='build',
                         help='Dependency type')
     return parser
@@ -19,9 +20,10 @@ def get_parser():
 def main():
     parser = get_parser()
     args = parser.parse_args()
-    major, minor, *_ = sys.version_info
-    assert major == 3
+    major, minor, *_ = args.py_ver.split('.')
+    assert major == "3"
     np_version = "1.22.2"
+    minor = int(minor)
     if minor >= 12:
         np_version = "1.26.0"
     elif minor >= 11: