|
8 | 8 | version="0.0.0"
|
9 | 9 |
|
10 | 10 | [tool.poetry.dependencies]
|
11 |
| - datasets=">=2.15.0" |
12 |
| - einops=">=0.6" |
13 |
| - pydantic=">=2.5.2" |
| 11 | + datasets=">=2.16.1" |
| 12 | + einops=">=0.7.0" |
| 13 | + pydantic=">=2.5.3" |
14 | 14 | python=">=3.10, <3.12"
|
15 | 15 | strenum=">=0.4.15"
|
16 |
| - tokenizers=">=0.15.0" |
17 |
| - torch=">=2.1.1" |
18 |
| - transformers=">=4.35.2" |
19 |
| - wandb=">=0.16.1" |
| 16 | + tokenizers=">=0.15.1" |
| 17 | + torch=">=2.1.2" |
| 18 | + transformers=">=4.37.1" |
| 19 | + wandb=">=0.16.2" |
20 | 20 | zstandard=">=0.22.0" # Required for downloading datasets such as The Pile
|
21 | 21 |
|
22 | 22 | [tool.poetry.group]
|
23 | 23 | [tool.poetry.group.dev.dependencies]
|
24 |
| - jupyter=">=1" |
25 |
| - plotly=">=5" |
26 |
| - poethepoet=">=0.24.2" |
27 |
| - pydoclint=">=0.3.8" |
28 |
| - pyright=">=1.1.340" |
29 |
| - pytest=">=7" |
30 |
| - pytest-cov=">=4" |
| 24 | + jupyter=">=1.0.0" |
| 25 | + plotly=">=5.18.0" |
| 26 | + poethepoet=">=0.24.4" |
| 27 | + pydoclint=">=0.3.9" |
| 28 | + pyright=">=1.1.348" |
| 29 | + pytest=">=7.4.4" |
| 30 | + pytest-cov=">=4.1.0" |
31 | 31 | pytest-integration=">=0.2.3"
|
32 | 32 | pytest-timeout=">=2.2.0"
|
33 | 33 | pytest-xdist="^3.5.0"
|
34 | 34 | ruff=">=0.1.14"
|
35 | 35 | syrupy=">=4.6.0"
|
36 | 36 |
|
37 | 37 | [tool.poetry.group.demos.dependencies]
|
38 |
| - huggingface-hub=">=0.19.4" |
| 38 | + huggingface-hub=">=0.20.3" |
39 | 39 | ipywidgets=">=8.1.1"
|
40 |
| - jupyterlab=">=3" |
41 |
| - transformer-lens=">=1.9.0" |
| 40 | + jupyterlab=">=4.0.11" |
| 41 | + transformer-lens=">=1.13.0" |
42 | 42 |
|
43 | 43 | [tool.poetry.group.docs.dependencies]
|
44 | 44 | mkdocs=">=1.5.3"
|
45 | 45 | mkdocs-gen-files=">=0.5.0"
|
46 | 46 | mkdocs-htmlproofer-plugin=">=1.0.0"
|
47 | 47 | mkdocs-literate-nav=">=0.6.1"
|
48 |
| - mkdocs-material=">=9.4.10" |
| 48 | + mkdocs-material=">=9.5.5" |
49 | 49 | mkdocs-section-index=">=0.3.8"
|
50 | 50 | mkdocstrings={extras=["python"], version=">=0.24.0"}
|
51 |
| - mkdocstrings-python=">=1.7.3" |
| 51 | + mkdocstrings-python=">=1.8.0" |
52 | 52 | mknotebooks=">=0.8.0"
|
53 | 53 | pygments=">=2.17.2"
|
54 |
| - pymdown-extensions=">=10.5" |
55 |
| - pytkdocs-tweaks=">=0.0.7" |
| 54 | + pymdown-extensions=">=10.7" |
| 55 | + pytkdocs-tweaks=">=0.0.8" |
56 | 56 |
|
57 | 57 | [tool.poetry.scripts]
|
58 | 58 | join-sae-sweep='sparse_autoencoder.train.join_sweep:run'
|
|
0 commit comments