finish fixing the test

Author: aignas

MODULE.bazel

@@ -79,7 +79,7 @@ pip = use_extension("//python/extensions:pip.bzl", "pip")
             "cp{major}{minor}",
         ],
         whl_platform_tags = [
-            "linux_*_{}".format(cpu),
+            "linux_{}".format(cpu),
             "manylinux_*_{}".format(cpu),
         ],
     )

python/private/pypi/select_whl.bzl

@@ -8,6 +8,7 @@ _ANDROID = "android"
 _ANY = "any"
 _IOS = "ios"
 _MANYLINUX = "manylinux"
+_MACOSX = "macosx"
 _MUSLLINUX = "musllinux"
 
 def _value_priority(*, tag, values):
@@ -27,10 +28,11 @@ def _platform_tag_priority(*, tag, values):
     # https://packaging.python.org/en/latest/specifications/platform-compatibility-tags/
 
     if not (
-        tag.startswith(_MANYLINUX) or
-        tag.startswith(_MUSLLINUX) or
         tag.startswith(_ANDROID) or
-        tag.startswith(_IOS)
+        tag.startswith(_IOS) or
+        tag.startswith(_MACOSX) or
+        tag.startswith(_MANYLINUX) or
+        tag.startswith(_MUSLLINUX)
     ):
         res = _value_priority(tag = tag, values = values)
         if res == None:

tests/pypi/extension/extension_tests.bzl

@@ -410,8 +410,8 @@ def _test_torch_experimental_index_url(env):
                         whl_platform_tags = whl_platform_tags,
                     )
                     for (os, cpu), whl_platform_tags in {
-                        ("linux", "x86_64"): ["linux_*_x86_64", "manylinux_*_x86_64"],
-                        ("linux", "aarch64"): ["linux_*_aarch64", "manylinux_*_aarch64"],
+                        ("linux", "x86_64"): ["linux_x86_64", "manylinux_*_x86_64"],
+                        ("linux", "aarch64"): ["linux_aarch64", "manylinux_*_aarch64"],
                         ("osx", "aarch64"): ["macosx_*_arm64"],
                         ("windows", "x86_64"): ["win_amd64"],
                         ("windows", "aarch64"): ["win_arm64"],  # this should be ignored