Enable Python 3.15 / 3.15t for macOS wheel builds

Extend the opt-in preview Python versions (INCLUDE_PREVIEW_PYTHON_VERSIONS) to
macOS arm64 so 3.15 / 3.15t are generated for macOS wheels, matching the Linux
enablement. The interpreter setup (conda-forge python_dev/python_rc labels) is
shared across OSes via .github/actions/setup-binary-builds. 3.15 / 3.15t are
left out of MACOS_PYTHON_POINT_VERSIONS so they flow to that shared case as
"3.15" / "3.15t" and resolve to the pinned conda-forge beta.

The macOS base env ships a newer conda (>=26, required by its py3.14 base) that
returns HTTP 400 downloading the 3.15 pre-release .conda from conda-forge label
subchannels. For 3.15* on macOS, downgrade the base env and pin the older conda
used on Linux/Windows, which downloads them correctly (same approach used for
earlier macOS pre-release bring-ups).

Also update the matrix unit tests (macOS arm64 now opts in).

Authored with assistance from Claude Code (AI assistant).

Author: atalman

.github/actions/setup-binary-builds/action.yml

@@ -138,6 +138,18 @@ runs:
           # untouched.
           if [[ "${RUNNER_OS}" != "macOS" ]]; then
             conda install -y conda=25.3.0
+          elif [[ "${PYTHON_VERSION}" == 3.15* ]]; then
+            # The macOS base env ships a newer conda (>=26, required by its
+            # py3.14 base) that returns HTTP 400 when downloading the 3.15
+            # pre-release .conda files from conda-forge label subchannels.
+            # Downgrade the base env and pin the older conda used on
+            # Linux/Windows, which downloads them correctly (same approach used
+            # for earlier macOS pre-release bring-ups).
+            conda install -y python=3.9
+            if [[ -n "$(conda list | grep conda-anaconda-telemetry)" ]]; then
+              conda uninstall -y conda-anaconda-telemetry conda-anaconda-tos
+            fi
+            conda install -y conda=24.7.1 conda-libmamba-solver=24.1.0
           fi
 
           # Pin openssl=3.5.6 by default: openssl 3.5.7 regressed ASN.1 parsing

tools/scripts/generate_binary_build_matrix.py

@@ -448,15 +448,21 @@ def generate_wheels_matrix(
 ) -> List[Dict[str, str]]:
     package_type = "wheel"
 
+    explicitly_requested_versions = bool(python_versions)
+
     if not python_versions:
         # Define default python version
         python_versions = list(PYTHON_ARCHES)
 
         # Opt-in preview versions (e.g. 3.15/3.15t) are torch-only and validated
-        # on Linux x86 and aarch64 only. Append them to the default list so the
-        # shared default (used by domain libraries, Windows and macOS) is
+        # on Linux x86/aarch64 and macOS arm64. Append them to the default list
+        # so the shared default (used by domain libraries and Windows) is
         # unaffected.
-        if include_preview_python_versions and os in (LINUX, LINUX_AARCH64):
+        if include_preview_python_versions and os in (
+            LINUX,
+            LINUX_AARCH64,
+            MACOS_ARM64,
+        ):
             python_versions += PREVIEW_PYTHON_ARCHES_DICT.get(channel, [])
 
     if os == WINDOWS_ARM64:

tools/tests/test_generate_binary_build_matrix.py

@@ -149,11 +149,11 @@ def _test_channel_python_versions(
         )
         return {entry["python_version"] for entry in out["include"]}
 
-    def test_preview_python_versions_opt_in_on_linux(self):
-        # 3.15 / 3.15t are validated on Linux x86 and aarch64 for the nightly and
-        # test channels when explicitly opted in.
+    def test_preview_python_versions_opt_in(self):
+        # 3.15 / 3.15t are validated on Linux x86/aarch64 and macOS arm64 for the
+        # nightly and test channels when explicitly opted in.
         for channel in ("nightly", "test"):
-            for operating_system in ("linux", "linux-aarch64"):
+            for operating_system in ("linux", "linux-aarch64", "macos-arm64"):
                 versions = self._test_channel_python_versions(
                     operating_system, include_preview="enable", channel=channel
                 )
@@ -167,14 +167,13 @@ def test_preview_python_versions_off_by_default(self):
             self.assertNotIn("3.15", versions)
             self.assertNotIn("3.15t", versions)
 
-    def test_preview_python_versions_excluded_on_non_linux(self):
-        # Windows and macOS must not pick up the preview versions even when opted in.
-        for operating_system in ("windows", "macos"):
-            versions = self._test_channel_python_versions(
-                operating_system, include_preview="enable"
-            )
-            self.assertNotIn("3.15", versions)
-            self.assertNotIn("3.15t", versions)
+    def test_preview_python_versions_excluded_on_windows(self):
+        # Windows must not pick up the preview versions even when opted in.
+        versions = self._test_channel_python_versions(
+            "windows", include_preview="enable"
+        )
+        self.assertNotIn("3.15", versions)
+        self.assertNotIn("3.15t", versions)
 
     def test_preview_python_versions_excluded_on_release_channel(self):
         # Preview versions are defined for the nightly and test channels only,