NodeLens/pyproject.toml at develop · KempnerInstitute/NodeLens · 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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "nodelens"
version = "0.2.0"
description = "Node and channel metrics for neural network interpretability, importance, and intervention studies."
authors = [
    { name = "Research and Engineering at Kempner Institute", email = "kempner-research-engineering@g.harvard.edu" },
]
readme = "README.md"
requires-python = ">=3.8"
license = "MIT"
license-files = ["LICENSE"]
dependencies = [
    "numpy",
    "torch",
    "torchvision",
    "omegaconf",
    "seaborn",
    "matplotlib",
    "tqdm",
    "pandas",
    "scipy",
]

[tool.setuptools]
package-dir = { "" = "src" }

[tool.setuptools.packages.find]
where = ["src"]
include = ["nodelens*"]


[project.optional-dependencies]
dev = [
    "ruff",
    "mypy>=1.0,<1.17",
    "black>=24.0",
    "isort>=5.12,<6.1",
    "pytest",
    "pytest-sphinx",
    "twine>=1.11.0",
    "setuptools",
    "wheel",
    "build",
]
train = [
    "wandb",
    "beaker-gantry",
    "click",
    "torchmetrics",
    "smashed[remote]>=0.21.1",
    "safetensors",
    "datasets",
    "scikit-learn",
    "msgspec>=0.14.0",
]
all = [
    "nodelens[dev,train]",
]

docs = [
    "sphinx",
    "sphinx-rtd-theme",
    "sphinx-autodoc-typehints",
    "sphinxcontrib-napoleon",
    "myst-parser",
    "nbsphinx",
    "recommonmark",
    "sphinx-copybutton",
    "sphinxcontrib-bibtex",

]


[project.urls]
Homepage = "https://github.com/KempnerInstitute/nodelens"
Repository = "https://github.com/KempnerInstitute/nodelens"
Documentation = "https://github.com/KempnerInstitute/nodelens#readme"
Artifacts = "https://huggingface.co/datasets/hsafaai/supernodes-scar-artifacts"

[tool.black]
line-length = 150

[tool.isort]
profile = "black"
line_length = 150

[tool.ruff]
line-length = 150

[tool.ruff.lint]
extend-ignore = ["E402", "E721", "E722"]

[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["F401"]  # Allow unused imports in __init__ files (used for exports)
"src/nodelens/external/**" = ["E721", "E722"]  # Allow in external code

[tool.pytest.ini_options]
testpaths = ["tests"]
markers = [
    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
    "integration: marks tests as integration tests",
    "gpu: marks tests that require GPU",
    "paper_critical: marks tests for paper-critical modules",
]