KernelGYM/requirements.txt at main · hkust-nlp/KernelGYM · 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
anthropic
modal
numpy
openai
packaging
pydra_config
tqdm
datasets
transformers
google-generativeai
together
pytest
ninja
archon-ai
einops
pydra-config==0.0.15
dotenv

# Core dependencies
# fastapi==0.104.1
uvicorn[standard]==0.24.0
# pydantic==2.5.0
pydantic-settings==2.10.1
# python-multipart==0.0.6c
python-multipart==0.0.6

# Task queue
celery==5.3.4
redis==5.0.1
kombu==5.3.4

# Database and caching
sqlalchemy==2.0.23
alembic==1.12.1

# Monitoring and logging
# prometheus-client==0.19.0
structlog==15.1.0
# rich==13.7.0

# Security
# cryptography==41.0.8
python-jose[cryptography]==3.3.0
passlib[bcrypt]==1.7.4

# GPU and ML
torch>=2.0.0
torchvision>=0.15.0
nvidia-ml-py3==7.352.0

# Testing
pytest==7.4.3
pytest-asyncio==0.21.1
# httpx==0.25.2
pytest-mock==3.12.0

# Development
black==23.11.0
isort==5.12.0
flake8==6.1.0
pre-commit==3.5.0

# Utils
python-dotenv==1.0.0
click==8.1.7
tenacity==8.2.3
psutil==5.9.6
aiohttp==3.9.1