language/pyproject.toml at main · brain-score/language · GitHub

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
################################################################
#### project config ####
################################################################

[project]
name = "brainscore_language"
version = "2.2.1"
description = ""
authors = []
license = { 'file' = 'LICENSE' }
readme = "README.md"
requires-python = ">=3.11"

dependencies = [
    "tqdm",
    "numpy <2",
    "brainscore-core",
    "fire",
    "scikit-learn >=1.6", # for metric_helpers/transformations.py cross-validation
    "pandas <3",
    # model_helpers dependencies
    "torch >=1.9.1",
    "transformers >=4.11.3",
    "gensim",
    "joblib",
    # submission dependencies
    "requests"
]

[project.optional-dependencies]
test = [
    "pytest",
    "pytest-check <=1.3.0",
]

[build-system]
requires = [
    "setuptools >=65",
    "wheel"
]


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

[tool.mypy]
ignore_missing_imports = true


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

filterwarnings = [
    "ignore::DeprecationWarning",
    "ignore::FutureWarning"
]

markers = [
    "memory_intense",
    "travis_slow",
    "slow",
    "private_access",
]


################################################################
#### setuptools packaging config ####
################################################################

[tool.setuptools.package-data]
# include non-python files (e.g. 'requirements.txt') in package install
"brainscore_language.benchmarks" = ["**"]
"brainscore_language.data" = ["**"]
"brainscore_language.metrics" = ["**"]
"brainscore_language.models" = ["**"]