sp26-gke-research/pixi.toml at main · mlberkeley/sp26-gke-research · 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
[workspace]
name = "sp26-google-gke-research"
channels = ["conda-forge"]
platforms = ["linux-64", "linux-aarch64", "osx-arm64", "osx-64"]

[tasks]
postinstall = "pip install --no-build-isolation --no-deps --disable-pip-version-check -v -e ."

geodata-all = "python -m sp26_gke.geodata.cli run-all"
geodata-download-pois = "python -m sp26_gke.geodata.cli download-pois"
geodata-download-census = "python -m sp26_gke.geodata.cli download-census"
geodata-preprocess = "python -m sp26_gke.geodata.cli preprocess"

geodata-plot = "python sp26_gke/plotting/geodata/generate_figures.py"
gke-dummy-job = "python -m sp26_gke.workflows.gke_dummy_job"

[dependencies]
python = ">=3.12,<3.14"
pip = "*"

[feature.py_utilities.dependencies]
typer = "*"
colorama = "*"
tqdm = "*"
python-dotenv = "*"
pyyaml = "*"
structlog = "*"
pydantic = ">=2"

[feature.py_data_processing.dependencies]
pyarrow = "*"
pandas = "*"
polars = "*"
numpy = "*"


[feature.py_ml.dependencies]
pytorch = ">=2.1.0"
torchvision = ">=0.16.0"
scikit-learn = "*"
scipy = "*"

[feature.py_plotting.dependencies]
matplotlib = "*"
seaborn = "*"
plotly = "*"


[feature.py_dev.dependencies]
ipython = "*"
ipykernel = "*"
nbconvert = "*"

[feature.py_test.dependencies]
mypy = ">=1.13"
pytest = ">=6"
pytest-cov = "*"

[feature.py_test.tasks]
test = "pytest tests/"

[feature.py_lint.dependencies]
docformatter = "*"
pre-commit = "*"
pre-commit-hooks = "*"
prettier = "*"
ruff = "*"
taplo = "*"
typos = "*"

[feature.py_lint.tasks]
pre-commit-install = "pre-commit install"
pre-commit-run = "pre-commit run -a"

[feature.infra.dependencies]
google-cloud-sdk = "*"
terraform = "*"
go-sops = "*"
age = "*"
kubernetes-client = "*"

[environments]
default = { features = [
  "infra",
  "py_utilities",
  "py_data_processing",
  "py_ml",
  "py_plotting",
  "py_dev",
  "py_test",
  "py_lint",
] }
lint = { features = ["py_lint"], no-default-feature = true }
infra = { features = ["infra"], no-default-feature = true }