-
-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
pyproject.toml
57 lines (51 loc) · 1.36 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
[build-system]
requires = ['setuptools>=61']
build-backend = 'setuptools.build_meta'
[project]
name = 'segmentation_models_pytorch'
description = 'Image segmentation models with pre-trained backbones. PyTorch.'
readme = 'README.md'
requires-python = '>=3.9'
license = {file = 'LICENSE'}
authors = [{name = 'Pavel Iakubovskii', email = '[email protected]'}]
classifiers = [
'License :: OSI Approved :: MIT License',
'Programming Language :: Python',
'Programming Language :: Python :: 3',
'Programming Language :: Python :: Implementation :: CPython',
'Programming Language :: Python :: Implementation :: PyPy',
]
dependencies = [
'efficientnet-pytorch==0.7.1',
'huggingface-hub>=0.24.6',
'numpy',
'pillow',
'pretrainedmodels==0.7.4',
'six',
'timm==0.9.7',
'torch',
'torchvision>=0.5',
'tqdm',
]
dynamic = ['version']
[project.optional-dependencies]
docs = [
'autodocsumm',
'huggingface-hub',
'six==1.15.0',
'sphinx<7',
'sphinx-book-theme==1.1.2',
]
test = [
'pytest',
'ruff==0.5.2',
]
[project.urls]
Homepage = 'https://github.com/qubvel-org/segmentation_models.pytorch'
[tool.ruff]
extend-include = ['*.ipynb']
fix = true
[tool.setuptools.dynamic]
version = {attr = 'segmentation_models_pytorch.__version__.__version__'}
[tool.setuptools.packages.find]
include = ['segmentation_models_pytorch*']