pyproject.toml

# project langbrainscore
# (C) Evlab, MIT BCS, 2021-2022


################################################################
#### poetry config ####
################################################################

[tool.poetry]
name = "langbrainscore"
version = "0.1.0"
description = ""
authors = ["Greta Tuckute <gretatu@mit.edu>", 
           "Aalok Sathe <asathe@mit.edu>", 
           "Ben Lipkin <lipkinb@mit.edu>"]
license = "MIT"
exclude = ["langbrainscore/data/"]

[tool.poetry.dependencies]
python = "3.8.*"
tqdm = "^4.62.3"
transformers = "4.11.3"
torch = "^1.9.1"
torchvision = "^0.10.1"
torchaudio = "^0.9.1"
numpy = "^1.21.2"
colorama = "^0.4.4"
pandas = "^1.3.4"
scikit-learn = "^1.0.1"
xarray = "^0.20.0"
matplotlib = "^3.5.1"
methodtools = "^0.4.5"
diskcache = "^5.4.0"
zarr = "^2.11.1"
wandb = "^0.12.11"
randomname = "^0.1.5"
pyarrow = "^8.0.0"
fastparquet = "^0.8.1"
black = "^22.3.0"
joblib = "^1.1.0"

[tool.poetry.dev-dependencies]
pdoc3 = "^0.10.0"
ipython = "^8.0.1"
mypy = "^0.931"
lxml = "^4.7.1"
pytest = "^7.0.0"
coverage = {extras = ["toml"], version = "^6.3.1"}
pytest-html = "^3.1.1"
yapf = "^0.32.0"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"



################################################################
#### mypy type-checking config ####
################################################################

[tool.mypy]
ignore_missing_imports = true


[tool.pytest.ini_options]
testpaths = [
    "tests"
]