-
Notifications
You must be signed in to change notification settings - Fork 357
/
pyproject.toml
128 lines (123 loc) · 3.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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
[build-system]
requires = ["setuptools>=68.0"]
build-backend = "setuptools.build_meta"
[project]
name = "lit-nlp"
# LINT.IfChange
version = "1.3"
# LINT.ThenChange(./lit_nlp/package.json)
authors = [
{ name="Google, LLC", email="lit-dev@google.com" }
]
description = "🔥LIT: The Learning Interpretability Tool"
readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.9"
# LINT.IfChange
dependencies = [
"absl-py>=1.4.0",
"annoy>=1.17.3",
"attrs>=22.1.0",
"etils[epath]>=1.5.0",
"filelock>=3.12.3",
"google-cloud-translate>=3.11.1",
"ipython>=7.34.0",
"Levenshtein>=0.21.1",
"matplotlib>=3.7.1",
"ml-collections>=0.1.1",
"numpy>=1.24.1,<2.0.0",
"pandas[output-formatting]>=2.0.3",
"Pillow>=10.0.0",
"portpicker>=1.5.2",
"requests>=2.31.0",
"rouge-score>=0.1.2",
"sacrebleu>=2.3.1",
"saliency>=0.1.3",
"scikit-learn>=1.0.2",
"scipy>=1.10.1",
"shap>=0.42.0,<0.46.0",
"six>=1.16.0",
"termcolor>=2.3.0",
"tqdm>=4.64.0",
"umap-learn>=0.5.1",
"werkzeug>=2.2.3",
]
# LINT.ThenChange(./requirements.txt)
classifiers = [
"Development Status :: 5 - Production/Stable",
"Framework :: Jupyter",
"Framework :: Pytest",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
"Programming Language :: JavaScript",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.10",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Information Analysis",
"Topic :: Scientific/Engineering :: Visualization",
"Typing :: Typed",
]
keywords = [
"interpretability",
"interpretable artifical intelligence",
"explainable",
"explainable artifical intelligence",
"XAI",
"computer vision",
"natural language processing",
"artifical intelligence",
"machine learning",
"deep learning",
"visualization",
"visual analytics",
"tensorflow",
"tf",
"torch",
]
[project.optional-dependencies]
# LINT.IfChange
examples-common = [
"gunicorn>=20.1.0",
"tensorflow>=2.16.0",
"transformers>=4.27.1",
]
# LINT.ThenChange(./requirements_examples_common.txt)
# LINT.IfChange
examples-discriminative-ai = [
"lit-nlp[examples-common]",
"tensorflow-datasets>=4.9.0",
"tf-keras>=2.16",
]
# LINT.ThenChange(./requirements_examples_discriminative_ai.txt)
# LINT.IfChange
examples-generative-ai = [
"lit-nlp[examples-common]",
"google-cloud-aiplatform>=1.60.0",
"keras>=3.0.0",
"keras-nlp>=0.14.0",
"sentencepiece==0.1.99",
"tensorflow-text>=2.16.0",
"torch>=2.0.0",
"vertexai>=1.49.0",
]
# LINT.ThenChange(./requirements_examples_generative_ai.txt)
# LINT.IfChange
test = [
"lit-nlp[examples-discriminative-ai]",
"lit-nlp[examples-generative-ai]",
"lime==0.2.0.1",
"pytest>=7.4.0,<8.0.0",
"webtest>=2.0",
]
# LINT.ThenChange(./requirements_test.txt)
[project.urls]
"Homepage" = "https://github.com/pair-code/lit"
"Repository" = "https://github.com/PAIR-code/lit"
"Bug Tracker" = "https://github.com/PAIR-code/lit/issues"
[tool.setuptools.packages.find]
include = ["lit_nlp*"]
exclude = ["*_test"]
[tool.pytest.ini_options]
addopts = "--import-mode=importlib"