-
Notifications
You must be signed in to change notification settings - Fork 7
/
pyproject.toml
110 lines (98 loc) · 4.12 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
[tool.poetry]
name = "intel-ai-safety"
version = "1.1.0"
description = ""
authors = ["IntelAI <IntelAI@intel.com>"]
readme = "README.md"
packages = []
[tool.poetry.dependencies]
python = ">=3.9, <3.11"
"intel-ai-safety-model-card-gen" = {path = "model_card_gen", develop = true, optional=true}
"intel-ai-safety-explainer" = {path = "explainer", develop = true, optional=true}
"intel-ai-safety-explainer-attributions" = {path = "plugins/explainers/attributions", develop = true, optional=true}
"intel-ai-safety-explainer-attributions-hugging-face" = {path = "plugins/explainers/attributions-hugging-face", develop = true, optional=true}
"intel-ai-safety-explainer-captum" = {path = "plugins/explainers/captum", develop = true, optional=true}
"intel-ai-safety-explainer-cam-tensorflow" = {path = "plugins/explainers/cam-tensorflow", develop = true, optional=true}
"intel-ai-safety-explainer-cam-pytorch" = {path = "plugins/explainers/cam-pytorch", develop = true, optional=true}
"intel-ai-safety-explainer-metrics" = {path = "plugins/explainers/metrics", develop = true, optional=true}
[tool.poetry.group.test]
optional = true
[tool.poetry.group.test.dependencies]
backcall = '0.2.0'
datasets = '2.14.4'
deepdiff = '6.7.1'
intel-tensorflow = '2.14.0'
pickleshare = '0.7.5'
pillow = '10.3.0'
pytest = '8.1.1'
scikit-learn = '1.5.0'
sentencepiece = '0.2.0'
tensorflow-hub = '0.15.0'
torch = {version = "2.2.0", source = "pytorch-cpu"}
torchvision = {version = "0.17.0", source = "pytorch-cpu"}
[tool.poetry.extras]
model-card = ["intel-ai-safety-model-card-gen"]
explainer = ["intel-ai-safety-explainer"]
explainer-tensorflow = ["intel-ai-safety-explainer",
"intel-ai-safety-explainer-attributions",
"intel-ai-safety-explainer-cam-tensorflow",
"intel-ai-safety-explainer-metrics",
]
explainer-pytorch = ["intel-ai-safety-explainer",
"intel-ai-safety-explainer-captum",
"intel-ai-safety-explainer-cam-pytorch",
"intel-ai-safety-explainer-metrics"
]
explainer-all = ["intel-ai-safety-explainer",
"intel-ai-safety-explainer-attributions",
"intel-ai-safety-explainer-attributions-hugging-face",
"intel-ai-safety-explainer-captum",
"intel-ai-safety-explainer-cam-tensorflow",
"intel-ai-safety-explainer-cam-pytorch",
"intel-ai-safety-explainer-metrics",
]
all = ["intel-ai-safety-model-card-gen",
"intel-ai-safety-explainer",
"intel-ai-safety-explainer-attributions",
"intel-ai-safety-explainer-attributions-hugging-face",
"intel-ai-safety-explainer-captum",
"intel-ai-safety-explainer-cam-tensorflow",
"intel-ai-safety-explainer-cam-pytorch",
"intel-ai-safety-explainer-metrics"
]
[[tool.poetry.source]]
name = "PyPI"
priority = "primary"
[[tool.poetry.source]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"
[[tool.poetry_bumpversion.replacements]]
files = [
"model_card_gen/pyproject.toml",
"explainer/pyproject.toml",
"plugins/explainers/attributions/pyproject.toml",
"plugins/explainers/attributions-hugging-face/pyproject.toml",
"plugins/explainers/captum/pyproject.toml",
"plugins/explainers/cam-tensorflow/pyproject.toml",
"plugins/explainers/cam-pytorch/pyproject.toml",
"plugins/explainers/metrics/pyproject.toml",
]
[[tool.poetry_bumpversion.replacements]]
files = ["intel_ai_safety/version.py",
"explainer/intel_ai_safety/explainer/version.py",
"model_card_gen/intel_ai_safety/model_card_gen/version.py"]
search = '__version__ = "{current_version}"'
replace = '__version__ = "{new_version}"'
[build-system]
requires = ["poetry"]
build-backend = "poetry.core.masonry.api"
[tool.black]
line-length = 120
extend-exclude = '''
/(\.git|\.venv|__pycache__|\.cache|\.tox|\.pytest_cache|\.mypy_cache|)/
\.pyc$
'''
force-exclude = '''docker/helm-runner/third_party|model_card_gen/intel_ai_safety/model_card_gen/model_card.py'''
preview = true
target-version = ["py39", "py310"]