From ff1117d293595079ccc31df7a2770f464d5532fc Mon Sep 17 00:00:00 2001
From: Jiaming Yuan <jm.yuan@outlook.com>
Date: Thu, 16 Oct 2025 05:25:17 +0800
Subject: [PATCH] Add nccl as dep for aarch64.

---
 ops/script/pypi_variants.py   | 2 +-
 python-package/pyproject.toml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/ops/script/pypi_variants.py b/ops/script/pypi_variants.py
index c18b11a472c4..0812f9315ecb 100644
--- a/ops/script/pypi_variants.py
+++ b/ops/script/pypi_variants.py
@@ -11,7 +11,7 @@
 STUB_IN_PATH = os.path.join(PY_PACKAGE, "pyproject.toml.stub.in")
 OUT_PATH = os.path.join(PY_PACKAGE, "pyproject.toml")
 
-NCCL_WHL = """    \"nvidia-nccl-{0} ; platform_system == 'Linux' and platform_machine != 'aarch64'\","""
+NCCL_WHL = """    \"nvidia-nccl-{0} ; platform_system == 'Linux'\","""
 
 NAME = "{{ name }}"
 NCCL = "{{ nccl }}"
diff --git a/python-package/pyproject.toml b/python-package/pyproject.toml
index af4230ea2835..93b48e41bfa3 100644
--- a/python-package/pyproject.toml
+++ b/python-package/pyproject.toml
@@ -33,7 +33,7 @@ classifiers = [
 dependencies = [
     "numpy",
     "scipy",
-    "nvidia-nccl-cu12 ; platform_system == 'Linux' and platform_machine != 'aarch64'",
+    "nvidia-nccl-cu12 ; platform_system == 'Linux'",
 ]
 
 [project.urls]