numpy>=1.26
pyyaml>=6.0
scikit-learn>=1.7
scipy>=1.14
statsmodels>=0.14
pandas>=2.2
pyarrow>=14.0
huggingface-hub>=0.24
certifi>=2024.0

[all]
replicaverse[dev,docs,information-value,kenlm,lens,lmer-rs,mmc,r,similarity-surprisal,test,viz,zuco]

[character-surprisal]
replicaverse[models]
genlm-bytes>=0.2.0

[dev]
replicaverse[character-surprisal,frequency,mmc,models,readers,test,typecheck,zuco]
ruff==0.15.14

[docs]
quartodoc>=0.11

[frequency]
wordfreq[cjk]>=3.0

[information-value]
spacy>=3.7
scipy>=1.10
nltk>=3.9

[kenlm]
kenlm>=0.2
mosestokenizer>=1.2

[lens]
replicaverse[models]
tuned-lens>=0.2

[lmer-rs]
lme_python>=0.2.5
polars>=1.0
pandas>=2.0

[mmc]
langdetect>=1.0.9

[models]
transformers<5,>=4.45
torch>=2.4
accelerate>=1.0

[r]
rpy2<3.6,>=3.5.11

[readers]
pyreadr>=0.5
nibabel>=5.0
openpyxl>=3.1
xlrd>=2.0.1

[similarity-surprisal]
flair>=0.13
sacremoses>=0.1.1

[test]
pytest
pytest-xdist
pytest-cov
pytest-socket>=0.7
sacremoses>=0.1.1
nltk>=3.9

[typecheck]
mypy>=1.10

[viz]
matplotlib>=3.8
pillow>=10.0

[zuco]
h5py>=3.10
