GeoGraph/pyproject.toml at main · instadeepai/GeoGraph · 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
[project]
name = "GeoGraph"
version = "0.1.0"
description = "Repository for inference of the GeoGraph model."
readme = "README.md"
requires-python = ">=3.12"

dependencies = [
    "datasets>=4.0.0",
    "hydra-core>=1.3.2",
    "numpy>=2.2.6",
    "seaborn>=0.13.2",
    "torch>=2.7.0",
    "transformers>=4.54.1",
    "zenodo-get>=2.0.0",
    "scipy",
    "matplotlib",
    "ipykernel>=6.29.5",
    "jupyter",
    "ipython",
    "tqdm",
    "pandas",
    "mdtraj>=1.9.7",
    "loguru>=0.7.3",
    "huggingface-hub>=0.36.0",
]

[dependency-groups]
dev = ["pre-commit>=3.7.1"]
test = ["pytest>=8.3.3"]

[tool.uv]
default-groups = ["dev", "test"]
package = true

# build system
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = [
    "src/geograph",
]

[tool.hatch.metadata]
allow-direct-references = true

## CONFIGURATION
# ruff
[tool.ruff]
fix = true
show-fixes = true
line-length = 120
target-version = "py312"
src = ["src", "tests"]

[tool.ruff.format]
indent-style = "space"
quote-style = "double"

[tool.ruff.lint]
logger-objects = ["loguru.logger"]
select = [
    "I",    # isort
    "E",    # pycodestyle (errors)
    "W",    # pycodestyle (warnings)
    "N",    # pep8-naming
    "D",    # pydocstyle
	"F",    # pyflakes (imports)
]
ignore = []

[tool.ruff.lint.pydocstyle]
convention = "google"

# mypy
[tool.mypy]
python_version = "3.12"
namespace_packages = true
incremental = false
cache_dir = ".mypy_cache"
warn_redundant_casts = true
warn_return_any = true
warn_unused_configs = true
warn_unused_ignores = false
allow_redefinition = true
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
check_untyped_defs = true
strict_optional = true
strict_equality = true
explicit_package_bases = true
follow_imports = "skip"
ignore_missing_imports = true

# pytest
[tool.pytest.ini_options]
addopts = [
    "tests",
    "-rxXs",
    "--color=yes",
    "--durations=0",
    "--strict-markers",
    "--doctest-modules",
]