pyproject.toml

[project]
name = "pytorch-sparse-utils"
description = "Low-level utilities and operations for Pytorch sparse tensors"
authors = [
    {name = "Matthew A. Wright", email = "mawright@lbl.gov"}
]
readme = "README.md"
dynamic = ["version"]
requires-python = ">= 3.9"
dependencies = [
    "numpy",
    "torch",
    "sparse",
]

[tool.setuptools.packages.find]
include = [
    "pytorch_sparse_utils",
    "pytorch_sparse_utils.*",
]

[project.optional-dependencies]
tests = [
    "pytest",
    "pytest-env",
    "hypothesis",
    "pytest-cov",
    "pytest-xdist",
]
minkowskiengine = ["MinkowskiEngine"]
spconv = ["spconv"]
docs = ["mkdocs", "mkdocstrings", "mkdocstrings-python"]

[tool.pytest.ini_options]
env = [
    "OMP_NUM_THREADS=16",  # MinkowskiEngine issues a warning on import if this is unset
    "CUDA_LAUNCH_BLOCKING=1",  # Makes debugging easier
]
markers = [
    "cuda_if_available: mark test to run on CUDA when available, otherwise fall back to CPU",
    "cpu_and_cuda: mark test to run on both cpu as well as CUDA if available"
]
filterwarnings = [
    "ignore:.*custom_[fb]wd.*:FutureWarning:spconv.pytorch.functional",  # spconv import
    "ignore:.*Importing from timm\\.models\\.layers.*:FutureWarning:timm.models.layers", 
    "ignore:.*locale.getdefaultlocale.*:DeprecationWarning",
]

[tool.coverage.run]
source = ["pytorch_sparse_utils"]
parallel = true

[tool.coverage.report]
exclude_also = [
    "def __repr__",
    "raise AssertionError",
    "raise NotImplementedError",
    "if __name__ == .__main__.:",
    "if TYPE_CHECKING:",
    "if torch\\.jit\\.is_scripting\\(\\):",
]