ICX360/pyproject.toml at main · IBM/ICX360 · 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
[project]
name            = "icx360"
version         = "0.1.0"
description     = "In-Context Explainability 360 Toolkit"
readme          = "README.md"
license         = { text = "Apache-2.0" }

authors         = [{ name = "IBM Research" }]
maintainers     = [
    { name = "ibm", email = "ibm@ibm.com" }
]

requires-python = ">=3.11,<3.13"

# core
dependencies = [
	"numpy",
	"transformers",
	"torch>=2.0",
	"jupyterlab",
	"notebook",
	"evaluate",
	"bert_score",
	"nltk",
	"sentencepiece",
	"protobuf",
	"toma",
	"sentence-transformers",
	"spacy",
	"pytest",
	"python-dotenv>=1.1.0,<2.0.0",
	"openai>=1.86.0,<2.0.0",
	"datasets<4.0.0",
    "accelerate",
	"pre-commit"
]

[tool.setuptools.packages]
find = {}

[tool.pytest.ini_options]
markers = [
    "slow: marks tests as slow (deselect with '-m \"not slow\"')",
	"vllm: marks tests as requiring a VLLM server (deselect with '-m \"not vllm\"')",
]

# optional groups
[project.optional-dependencies]
docs = [
    "mkdocs-material",
    "mkdocs-jupyter",
    "mkdocs-click",
    "mkdocs-include",
    "mkdocstrings[python]",
    "griffe_inherited_docstrings",
    "griffe-pydantic",
    "mkdocs-awesome-nav",
    "mkdocs-include-markdown-plugin",
]