oodeel/setup.cfg at master · deel-ai/oodeel · 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
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
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
[flake8]
# Use black line length:
max-line-length = 88
extend-ignore =
    # See https://github.com/PyCQA/pycodestyle/issues/373
    E203
per-file-ignores =
	*/__init__.py: F401

[tool:ruff]
line-length = 88

[mypy]
warn_return_any = True
warn_unused_configs = True
namespace_packages = True
ignore_missing_imports = True

[mypy-deel.datasets.*]
ignore_missing_imports = True

[mypy-PIL]
ignore_missing_imports = True

[mypy-tdqm]
ignore_missing_imports = True

[mypy-mnist]
ignore_missing_imports = True

[mypy-scipy]
ignore_missing_imports = True

[coverage:report]
exclude_also =
    raise NotImplementedError
    except ImportError

[tox:tox]
envlist = py{39,310,311}-tf{211,213,214,215},py{39,310,311}-torch{113,200,22,24,26},py{39,310,311}-lint,py{39}-tf{215}-torch{202}-coverage

[testenv:py{39,310,311}-tf{211,213,214,215}]
deps =
    packaging
    scikit_learn
    scipy
    setuptools
    pytest
    kneed
    tqdm
    pandas
    seaborn
    requests
    datasets
    plotly == 5.15.0
    tf211: protobuf>=3.9.2,<3.20
    tf213: protobuf>=3.20.3,<4.21.0
    tf214: protobuf>=3.20.3,<4.21.0
    tf215: protobuf>=3.20.3,<4.21.0
    tf211: numpy~=1.23.5
    tf213: numpy~=1.24.3
    tf214: numpy~=1.24.3
    tf215: numpy~=1.26.0
    tf211: tensorflow ~= 2.11.0
    tf211: tensorflow_datasets
    tf211: tensorflow_probability ~= 0.19.0
    tf213: tensorflow ~= 2.13.0
    tf213: tensorflow_datasets
    tf213: tensorflow_probability ~= 0.21.0
    tf214: tensorflow ~= 2.14.0
    tf214: tensorflow_datasets
    tf214: tensorflow_probability ~= 0.22.0
    tf215: tensorflow ~= 2.15.0
    tf215: tensorflow_datasets
    tf215: tensorflow_probability ~= 0.23.0
install_command = uv pip install {opts} {packages}
setenv =
    DL_LIB = tensorflow
commands =
    py39,py310,py311: uv pip install faiss-cpu==1.7.4
    tf211: uv pip install --force-reinstall numpy==1.24.0
    tf213: uv pip install --force-reinstall numpy==1.24.0
    tf214: uv pip install --force-reinstall numpy==1.24.0
    tf215: uv pip install --force-reinstall numpy==1.26.0
    pytest tests/tests_tensorflow

[testenv:py{39,310,311}-torch{113,200,22,24,26}]
deps =
    packaging
    scikit_learn
    scipy
    setuptools
    pytest
    kneed
    tqdm
    protobuf>=3.20.3,<4.21.0
    pandas
    seaborn
    requests
    plotly == 5.15.0
    torch113: torch == 1.13.1+cpu
    torch113: torchvision == 0.14.1+cpu
    torch113: transformers == 4.43.0
    torch200: torch == 2.0.0+cpu
    torch200: torchvision == 0.15.0+cpu
    torch200: transformers == 4.49.0
    torch200: datasets==2.19.1
    torch22: torch == 2.2.0+cpu
    torch22: torchvision == 0.17.0+cpu
    torch22: transformers == 4.49.0
    torch24: datasets==2.19.1
    torch24: torch == 2.4.0+cpu
    torch24: torchvision == 0.19.0+cpu
    torch24: transformers == 4.49.0
    torch24: datasets==2.19.1
    torch26: torch == 2.6.0+cpu
    torch26: torchvision == 0.21.0+cpu
    torch26: transformers == 4.49.0
    torch26: datasets
install_command = uv pip install --extra-index-url https://download.pytorch.org/whl/cpu {opts} {packages}
setenv =
    DL_LIB = torch
commands =
    py39,py310,py311: uv pip install faiss-cpu==1.7.4
    torch113: uv pip install --force-reinstall numpy==1.24.0
    torch200: uv pip install --force-reinstall numpy==1.26.0
    torch22: uv pip install --force-reinstall numpy==1.26.0
    torch24: uv pip install --force-reinstall numpy==1.26.0
    torch26: uv pip install --force-reinstall numpy==1.26.0
    pytest tests/tests_torch

[testenv:py{39,310,311}-lint]
deps =
    black
    flake8
    flake8-black
    mypy
    git+https://github.com/numpy/numpy-stubs.git
commands =
    black --check --diff setup.py oodeel tests
    flake8 oodeel tests
    # mypy --namespace-packages oodeel tests

[testenv:py{310}-tf{215}-torch{22}-coverage]
deps =
    packaging
    faiss_gpu
    scikit_learn
    scipy
    setuptools
    pytest
    kneed
    tqdm
    protobuf>=3.20.3,<4.21.0
    coverage
    pandas
    seaborn
    plotly == 5.15.0
    torch22: torch == 2.2.0+cpu
    torch22: torchvision == 0.17.0+cpu
    torch22: transformers == 4.49.0
    torch22: datasets==2.19.1
    tf215: tensorflow ~= 2.15.0
    tf215: tensorflow_datasets
    tf215: tensorflow_probability ~= 0.23.0
    tf215: datasets
install_command = uv pip install --extra-index-url https://download.pytorch.org/whl/cpu {opts} {packages}
setenv =
    DL_LIB = both
commands =
    tf215: uv pip install --force-reinstall numpy==1.26.0
    coverage run --source oodeel -m pytest
    coverage xml