pyproject.toml

[build-system]
requires = [ "setuptools", "setuptools-scm"]
build-backend = "setuptools.build_meta"

[project]
name = "torch_harmonics"
authors = [
    { name="Boris Bonev" },
    { name="Thorsten Kurth" },
    { name="Max Rietmann" },
    { name="Mauro Bisson" },
    { name="Andrea Paris" },
    { name="Alberto Carpentieri" },
    { name="Massimiliano Fatica" },
    { name="Jean Kossaifi" },
    { name="Nikola Kovachki" },
    { name="Christian Hundt" },
]

maintainers = [
    { name="Boris Bonev", email = "bbonev@nvidia.com" },
    { name="Thorsten Kurth", email = "tkurth@nvidia.com" },
]

readme = "README.md"

dynamic = ["version"]

description = "Differentiable signal processing on the sphere for PyTorch."
requires-python = ">=3.9"

classifiers = [
    "Development Status :: 3 - Alpha",
    "Programming Language :: Python :: 3.9",
    "License :: OSI Approved :: BSD License",
    "Operating System :: OS Independent",
]

dependencies = [
    "torch>=2.6.0",
    "numpy>=1.22.4",
]


[tool.setuptools_scm]
fallback_version = "0.9.1a0"

[tool.setuptools.packages.find]
    include = ["torch_harmonics*"]

[project.optional-dependencies]
dev = [
    "pytest>=6.0.0",
    "pytest-cov>=3.0.0",
    "coverage>=6.5.0",
    "parameterized",
    "scipy>=1.9.0",
]
filter_basis = [
    "scipy>=1.9.0",
]
2d3ds = [
    "requests",
    "tarfile",
    "tqdm",
    "pillow",
    "h5py",
]

[tool.black]
line-length = 180

[tool.cibuildwheel]
# Build for Python 3.9, 3.10, 3.11, 3.12
build = ["cp39-*", "cp310-*", "cp311-*", "cp312-*"]
build-frontend = "pip; args: --no-build-isolation"

# Use manylinux for Linux wheels
manylinux-x86_64-image = "manylinux_2_28"
manylinux-i686-image = "manylinux_2_28"

# Test the built wheel
test-command = [
  "python -c 'import torch; print(torch._C._GLIBCXX_USE_CXX11_ABI)'",
  "python -c 'import torch_harmonics; print(\"torch_harmonics imported successfully\"); from torch_harmonics.disco import optimized_kernels_is_available; print(f\"optimized kernels are available: {optimized_kernels_is_available()}\")'"
]

# Repair wheels with auditwheel - exclude all system, PyTorch, and CUDA libraries
repair-wheel-command = "auditwheel repair --exclude 'libc.so*' --exclude 'libm.so*' --exclude 'libpthread.so*' --exclude 'libdl.so*' --exclude 'libutil.so*' --exclude 'libnsl.so*' --exclude 'libresolv.so*' --exclude 'libcrypt.so*' --exclude 'libnss.so*' --exclude 'libselinux.so*' --exclude 'libpcre.so*' --exclude 'libz.so*' --exclude 'libbz2.so*' --exclude 'liblzma.so*' --exclude 'libtinfo.so*' --exclude 'libncurses.so*' --exclude 'libreadline.so*' --exclude 'libhistory.so*' --exclude 'libform.so*' --exclude 'libmenu.so*' --exclude 'libpanel.so*' --exclude 'libgcc_s.so*' --exclude 'libstdc++.so*' --exclude 'libopenblas.so*' --exclude 'liblapack.so*' --exclude 'libgfortran.so*' --exclude 'libquadmath.so*' --exclude 'libtorch*.so*' --exclude 'libc10.so*' --exclude 'libcudart.so*' --exclude 'libcublas.so*' --exclude 'libcurand.so*' --exclude 'libcusparse.so*' --exclude 'libcufft.so*' --exclude 'libcusolver.so*' --exclude 'libnvrtc.so*' --exclude 'libnvjpeg.so*' --exclude 'libcudnn.so*' -w {dest_dir} {wheel}"

# Skip building for platforms we don't support
skip = ["*-win32", "*-win_amd64", "*-macosx_arm64", "*-musllinux*"]