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]