forked from EleutherAI/sae
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
43 lines (38 loc) · 1.02 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
[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 = [
"datasets",
"einops",
"jaxtyping",
"safetensors",
"simple-parsing",
"torch",
"transformers",
"triton",
]
version = "0.1.0"
[project.scripts]
sae = "sae.__main__:run"
[tool.pyright]
include = ["sae*"]
reportPrivateImportUsage = false
[tool.setuptools.packages.find]
include = ["sae*"]
[tool.ruff]
# Enable pycodestyle (`E`), Pyflakes (`F`), and isort (`I`) codes
# See https://beta.ruff.rs/docs/rules/ for more possible rules
select = ["E", "F", "I"]
# Same as Black.
line-length = 88
# 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