climatem/pyproject.toml at main · RolnickLab/climatem · 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
[tool.poetry]
name = "climatem"
version = "0.0.0"
description = "Emulating climate model data with causal representation learning."
authors = ["Sebastian Hickman <sebastian.hickman@mila.quebec>"]
license = "MIT"
readme = "README.md"

[tool.poetry.group.dev]
optional = true

# [tool.poetry.group.pypi]
# optional = true

#[tool.poetry.group.pypi.dependencies]
## Commented out as a reference/example
## Optional dependencies that need to be installed with poetry
## `poetry install --with pypi`

[tool.poetry.dependencies]
# Main dependencies that can be still installed with the command
# `pip install -e .` and `pip install -e . --no-index`
# Ideal for main project dependencies and those that need to be installed
# via DRAC pre-built wheels
#
# Strongly recommended to use ~ instead of ^ for better compatibility and
# reproducibility
#
# ex. requests = "~2.31.0"
python = ">=3.10,<3.12"
accelerate = "~1.0.1"
torch = "~2.5.0"
pytorch-lightning = "~2.5.0"
scipy = "~1.15.1"
cartopy = "~0.24.1"
xarray = "~2025.1.2"
matplotlib = "~3.8.0"
opencv-python = "~4.11.0.86"
seaborn = "~0.13.2"
omegaconf = "~2.3.0"
torchmetrics = "~1.6.1"
geopy = "~2.4.1"
tigramite = "^5.2.7.0"
cfgrib = "^0.9.15.0"
cftime = "^1.6.4.post1"
dask = "^2025.2.0"
scikit-learn = "^1.6.1"
jupyter-contrib-nbextensions = "^0.7.0"
notebook = "^7.3.2"
jupyterlab-widgets = "^3.0.13"
healpy = "^1.19.0"

[tool.poetry.group.dev.dependencies]
# Optional dependencies that need to be installed with poetry
# `poetry install --with dev`
pytest = "~8.2.2"
pre-commit = "^3.6.2"
pylint = "^3.2.7"
flake8 = "^7.1.1"
mccabe = "^0.7.0"
jupyter = "^1.1.1"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"


#
# Scripts
#
[tool.poetry.scripts]

#
# Tools
#
[tool.flake8]
max-line-length = 120
ignore = ["E203", "E266", "E501", "W503"]
docstring-convention = "numpy"
max-complexity = 20
per-file-ignores = []
exclude = [
    ".git",
    "migrations",
    "__pycache__"
]

[tool.pylint]
disable = "C0114,C0115,C0116,R0903,R1710,W1203"
max-line-length = 120
max-locals = 20
max-args = 16
max-attributes = 20

[tool.black]
line-length = 120
target-version = ["py311"]

[tool.isort]
profile = "black"

[tool.flynt]
line-length = 120
transform-concats = true
verbose = true

[tool.docformatter]
style = "numpy"
pre-summary-newline = true
wrap-descriptions = 120
wrap-summaries = 120
blank = false
exclude = []
recursive = true