astroml/requirements.txt at main · Martha-code-dev/astroml · 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
# ============================================================================
# Full GPU-capable training stack.
#
# Pair this with `requirements-cpu.txt` for CPU-only torch wheels, or with
# `requirements-minimal.txt` for the smallest set that lets the Hydra
# configuration system + dataframes import. See REQUIREMENTS.md for the
# decision tree.
# ============================================================================

# ── Core ML / experiment tracking ──────────────────────────────────────────
mlflow>=2.10.0
torch>=2.0.0
torch-geometric>=2.3.0
pytorch-lightning>=2.0.0

# ── Numerics / dataframes ──────────────────────────────────────────────────
numpy>=1.24
scipy>=1.11.0
scikit-learn>=1.3.0
pandas>=2.0
polars>=1.0

# ── Database / configuration ───────────────────────────────────────────────
sqlalchemy>=2.0
alembic>=1.12
psycopg2-binary>=2.9
pyyaml>=6.0
hydra-core>=1.3.0
omegaconf>=2.3.0

# ── Networking / ingestion ─────────────────────────────────────────────────
aiohttp>=3.9
aiohttp-sse-client>=0.2.1
stellar-sdk>=9.0.0
tenacity>=8.4.0

# Transitive constraint: pin starlette >= 1.0.1 to address PYSEC-2026-161
# (Host header path injection). mlflow / fastapi-style deps pull it in
# transitively; without this pin pip-audit flags the older resolver pick.
starlette>=1.0.1

# ── Observability ──────────────────────────────────────────────────────────
prometheus-client>=0.19.0

# ── Feature store ──────────────────────────────────────────────────────────
redis>=5.0.0
cachetools>=5.3.0
pyarrow>=14.0.0
fastparquet>=2024.2.0
networkx>=3.2.0
joblib>=1.3.0
tqdm>=4.66.0
click>=8.1.0
rich>=13.7.0

# ── Cloud storage / artifact management ──────────────────────────────────────
fsspec>=2024.2.0
s3fs>=2024.2.0
gcsfs>=2024.2.0

# ── Visualization ──────────────────────────────────────────────────────────
matplotlib>=3.7.0
seaborn>=0.12.0

# ── Dev / testing ──────────────────────────────────────────────────────────
pytest>=7.4.0
pytest-asyncio>=0.23
pytest-cov>=4.1.0
pytest-mock>=3.12.0
black>=23.11.0
flake8>=6.1.0
mypy>=1.7.0

# ── Notebooks ──────────────────────────────────────────────────────────────
jupyter>=1.0.0
notebook>=7.0.0
ipykernel>=6.26.0
pre-commit>=3.7.0
isort>=5.13.0
ruff>=0.4.0