Mars-Bench/pyproject.toml at main · kerner-lab/Mars-Bench · 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
[build-system]
requires = ["setuptools>=74.0.0", "wheel>=0.44.0"]
build-backend = "setuptools.build_meta"

[project]
name = "MarsBench"
version = "0.1.0"
description = "A benchmarking suite for composition in machine learning models"
readme = "README.md"
requires-python = ">=3.6"
license = {text = "MIT"}
authors = [
    {name = "MarsBench Authors"}
]
dependencies = [
    "torch>=2.4.0",
    "torchvision>=0.19.0",
    "pytorch-lightning>=2.4.0",
    "hydra-core>=1.3.2",
    "omegaconf>=2.1.0",
    "pandas>=2.2.2",
    "pillow>=10.2.0",
    "segmentation-models-pytorch>=0.5.0",
    "wandb>=0.18.5",
    "albumentations>=1.3.1",
    "lxml>=5.1.0",
    "effdet>=0.4.0",
    "torchmetrics>=1.7.0",
    "tensorboard>=2.15.0",
    "rasterio>=1.4.3",
    "opencv-python>=4.10",
    "tifffile>=2025.3.30",
    "transformers>=4.51.3",
]

[tool.setuptools]
packages = ["marsbench"]

[project.optional-dependencies]
dev = [
    "pytest>=8.3.3",
    "pytest-cov>=5.0.0",
    "black>=24.8.0",
    "flake8>=4.0.1",
    "isort>=5.10.0",
    "pre-commit>=3.8.0",
    "flake8-pyproject",
]

[tool.pytest]
testpaths = ["tests"]
python_files = "test_*.py"
filterwarnings = [
    "ignore::ResourceWarning",
    "ignore::DeprecationWarning",
    "ignore:The `srun` command is available on your system but is not used:UserWarning",
    "ignore:.*cuda initialization.*:UserWarning",
    "ignore:.*overflow encountered in exp.*:RuntimeWarning",
]

[tool.black]
line-length = 120
target-version = ["py36", "py37", "py38", "py39"]
exclude = '''
/(
    \.git
  | \.hg
  | \.mypy_cache
  | \.tox
  | \.venv
  | _build
  | buck-out
  | build
  | dist
  | legacy_code
)/
'''

[tool.isort]
profile = "black"
force_single_line = true
skip_glob = ["legacy_code/**"]

[tool.flake8]
max-line-length = 120
extend-ignore = ["E203", "W503", "E402", "F403"]
ignore-long-lines = "^\\s*#|^\\s*\"{3}|^\\s*'{3}"
per-file-ignores = "tests/*.py:F401"
exclude = [
    ".git",
    "__pycache__",
    "build",
    "dist",
    "legacy_code",
]

[tool.env]
NO_ALBUMENTATIONS_UPDATE = "1"