forked from Phylliida/sae-k-sparse-mamba
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpyproject.toml
59 lines (52 loc) · 1.37 KB
/
pyproject.toml
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
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"
[project]
name = "sae"
description = "Sparse autoencoders"
readme = "README.md"
requires-python = ">=3.10"
keywords = ["interpretability", "explainable-ai"]
license = {text = "MIT License"}
dependencies = [
"accelerate", # For device_map in from_pretrained
"datasets",
"einops",
"jaxtyping",
"safetensors",
"simple-parsing",
"torch",
"transformers",
"transformer_lens",
"triton",
"mamba_lens @ git+https://github.com/Phylliida/MambaLens.git",
# format/CI
"ruff",
"pyright",
"pytest",
"numpy<2.0.0", # 2.0 Unsupported by datasets
]
version = "0.1.0"
[project.scripts]
sae = "sae.__main__:run"
[project.optional-dependencies]
dev-local = [
"pre-commit",
]
[tool.pyright]
include = ["sae*"]
reportPrivateImportUsage = false
[tool.setuptools.packages.find]
include = ["sae*"]
[tool.ruff]
line-length = 88
[tool.ruff.lint]
# Enable pycodestyle (`E`), Pyflakes (`F`), and isort (`I`) codes
# See https://beta.ruff.rs/docs/rules/ for more possible rules
select = ["E", "F", "I"]
# Make jaxtyping usable
ignore = ["F722"]
# Avoid automatically removing unused imports in __init__.py files.
# Such imports will be flagged with a dedicated message suggesting
# that the import is either added to the module's __all__ symbol
ignore-init-module-imports = true