-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathpyproject.toml
78 lines (69 loc) · 2.19 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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
[tool.poetry]
name = "maze-transformer"
version = "0.2.0"
description = ""
authors = ["Michael Ivanitskiy <[email protected]>", "Dan Valentine <[email protected]>", "Rusheb Shah <[email protected]>", "Lucia Quirke <[email protected]>", "Can Rager <[email protected]>", "Alex Spies <[email protected]>", "Chris Mathwin <[email protected]>", "Tilman Rauker <[email protected]>", "Guillaume Corlouer <[email protected]>"]
readme = "README.md"
packages = [{include = "maze_transformer"}]
repository = "https://github.com/understanding-search/maze-transformer"
[tool.poetry.dependencies]
python = ">=3.10,<3.13"
# dataset
maze-dataset = "^0.5.6"
# transformers
torch = "^2.4.0"
transformer-lens = "^2.4.0"
# transformers = ">=4.34" # Dependency in transformer-lens 1.14.0
# utils
muutils = "^0.6.7"
zanj = "^0.3.1"
# wandb = "^0.13.5" # note: TransformerLens forces us to use 0.13.5
wandb = "^0.17.5"
fire = "^0.6.0"
typing-extensions = "^4.12.2"
# plotting
matplotlib = "^3.7.0"
plotly = "^5.13.1"
circuitsvis = "^1.39.1"
seaborn = "^0.12.2"
# notebooks
ipykernel = "^6.22.0"
jupyter = "^1.0.0"
# analysis
scipy = "^1.11.3"
scikit-learn = "^1.3.1"
[tool.poetry.group.dev.dependencies]
pytest = "^7.3.1"
pycln = "^2.4.0"
isort = "^5.13.2"
black = "^24.4.2"
pytest-mock = "^3.10.0"
pytest-cov = "^4.1.0"
coverage-badge = "^1.1.0"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.pytest.ini_options]
# Ignore numpy deprecation warnings triggered by muutils
filterwarnings = [
# Warning from muutils: https://github.com/mivanit/muutils/issues/1
"ignore:`np\\.\\w*` is a deprecated alias for:DeprecationWarning",
# Warning from matplotlib. Issue: https://github.com/matplotlib/matplotlib/issues/25244
"ignore:Deprecated call to `pkg_resources.declare_namespace:DeprecationWarning"
]
testpaths = "tests"
norecursedirs="maze_transformer/utils/test_helpers"
[[tool.poetry.source]]
name = "torch_cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"
[[tool.mypy.overrides]]
module = "fire"
ignore_missing_imports = true
[tool.isort]
profile = "black"
known_third_party = "wandb"
[tool.pyright]
exclude = [".venv"]
venvPath = "."
venv = ".venv"