sparse-transformer-layers/pyproject.toml at main · BouchardLab/sparse-transformer-layers · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
[project]
name = "sparse-transformer-layers"
description = "Transformer layers 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",
    "pytorch_sparse_utils @ git+https://github.com/mawright/pytorch-sparse-utils.git",
    "nd_rotary_encodings @ git+https://github.com/mawright/nd-rotary-encodings.git",
]

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

[project.optional-dependencies]
tests = [
    "pytest",
    "pytest-env",
    "pytest-xdist",
    "pytest-cov",
    "hypothesis",
]
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",
]