-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathrequirements-ci.txt
More file actions
39 lines (33 loc) · 1.08 KB
/
Copy pathrequirements-ci.txt
File metadata and controls
39 lines (33 loc) · 1.08 KB
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
# Test-only dependency set for GitHub Actions.
#
# requirements.txt installs the full training + serving stack, which on a free
# runner means pulling CUDA-enabled torch (~2.5 GB), wandb, uvicorn and the
# Groq client — none of which the unit tests touch. That install was the cause
# of the CI timeouts. This file lists only what `pytest tests/` actually
# imports; torch itself is installed separately from the CPU wheel index in
# the workflow (~200 MB instead of ~2.5 GB).
# Model / tensor ops (torch comes from the CPU index — see ci.yml)
open_clip_torch>=2.24.0
timm>=0.9.16
# Data & metrics
numpy>=1.26.0
pandas>=2.2.0
scikit-learn>=1.4.0
Pillow>=10.2.0
opencv-python-headless>=4.9.0
# Schema validation (api/schemas.py)
pydantic>=2.7.0
fastapi>=0.111.0
python-multipart>=0.0.9
# Config parsing
pyyaml>=6.0.1
# Optional at runtime (report generation degrades to a template fallback
# without it), but installed in CI so api.inference imports on the same code
# path as production.
groq>=0.5.0
# Test tooling
pytest>=8.0.0
pytest-asyncio>=0.23.0
pytest-cov>=5.0.0
httpx>=0.27.0
ruff>=0.4.0