vit-plasticity/pyproject.toml at main · ambroiseodt/vit-plasticity · 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
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "vitef"
version = "0.0.0"
authors = [
  { name="Ambroise Odonnat", email="ambroiseodonnattechnologie@gmail.com" },
]
description = "Vision Transformer Finetuning Benefits from Non-Smooth Components"
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
    "Programming Language :: Python :: 3",
    "License :: OSI Approved :: MIT License",
    "Operating System :: OS Independent",
]

dependencies = [
    "einops",
    "fire",
    "numpy",
    "omegaconf",
    "scikit-learn",
    "scipy",
    "torch",
    "torchvision",
    "transformers",
]

[project.optional-dependencies]
dev = [
    "ruff",
]

pretrained = [
    "accelerate",
    "huggingface_hub[hf_xet]",
    "tiktoken",
    "timm",
]

visu = [
    "imageio",
    "ipykernel",
    "ipywidgets",
    "matplotlib",
    "pymupdf",
    "seaborn",
]

[tool.setuptools.packages.find]
where = ["src"]

[project.urls]
"Homepage" = "https://github.com/ambroiseodt/vit-plasticity"
"Documentation" = "https://github.com/ambroiseodt/vit-plasticity#readme"
"Bug Tracker" = "https://github.com/ambroiseodt/vit-plasticity/issues"

[tool.ruff]
line-length = 120

[tool.ruff.lint]
select = [
  "E", "F", "W",  # Default rules
  "ANN001",       # Function arguments should annotate types
  "ANN201",       # Public function should annotate return type
  "B",            # Common bugs
  "DTZ",          # Timezone issues
  "I",            # Sort inputs
  "PLE",          # Common errors
  "RUF100",       # Remove unused noqa
  "UP",           # Modern syntax
]

[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["F401"]