File size: 2,255 Bytes
a3f0b6e
0ca9ca4
a3f0b6e
0ca9ca4
 
a3f0b6e
 
 
 
ce2c431
0ca9ca4
 
 
 
 
 
 
 
 
 
 
 
 
b40c62a
854e05e
 
 
0ca9ca4
 
 
 
 
 
 
 
 
a3f0b6e
0ca9ca4
 
 
 
 
 
 
 
 
a3f0b6e
 
 
 
 
 
 
 
 
 
 
 
b40c62a
a3f0b6e
 
 
 
 
 
 
 
b40c62a
a3f0b6e
36ed17a
03beb83
 
0ca9ca4
16d3463
 
 
b40c62a
 
 
 
 
4828471
a3f0b6e
36ed17a
 
a3f0b6e
0ca9ca4
854e05e
 
0ca9ca4
 
a3f0b6e
 
854e05e
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
[tool.poetry]
name = "pytorch_fastapi_project"
version = "0.1.0"
description = "Consolidated PyTorch and FastAPI project for AWS deployment and GHA testing"
authors = ["soutrik71 <soutrik.chowdhury@ab-inbev.com>"]
license = "Apache-2.0"
readme = "README.md"

[tool.poetry.dependencies]
python = ">=3.10,<3.11"
black = "24.8.0"
coverage = ">=7.6.1"
hydra-colorlog = "1.2.0"
hydra-core = "1.3.2"
lightning = {version = "2.4.0", extras = ["extra"]}
loguru = "0.7.2"
pytest = "^8.3.3"
rich = "13.8.1"
rootutils = "1.0.7"
tensorboard = "2.17.1"
timm = "1.0.9"
pandas = "^2.2.3"
numpy = "^1.26.0"
ruff = "*"
torch = {version = "^2.4.1", source = "pytorch_cuda"}
torchvision = {version = "^0.19.1", source = "pytorch_cuda"}
torchaudio = {version = "^2.4.1", source = "pytorch_cuda"}
seaborn = "^0.13.2"
pydantic = "^2.9.2"
kaggle = "^1.6.17"
pytest-cov = "^5.0.0"
pytest-mock = "^3.14.0"
flake8 = "^7.1.1"
dvc-gdrive = "^3.0.1"
dvc-azure = "^3.1.0"
transformers = "^4.45.2"
fastapi = "^0.115.4"
pydantic-settings = "^2.6.1"
uvicorn = "^0.32.0"
tenacity = "^9.0.0"
gunicorn = "^23.0.0"
aim = "^3.25.0"
mlflow = "^2.17.1"
hydra-optuna-sweeper = "^1.2.0"
dvc = "^3.56.0"
platformdirs = "3.10"
fastapi-utils = "^0.7.0"
httpx = "^0.27.2"
typing-inspect = "^0.9.0"
requests = "^2.32.3"
fastapi-restful = {extras = ["all"], version = "^0.6.0"}
aioredis = "^2.0.1"
psycopg2-binary = "^2.9.10"
asyncpg = "^0.30.0"
confluent-kafka = "^2.6.0"
aiokafka = "^0.12.0"
azure-servicebus = "^7.12.3"
aiohttp = "^3.10.10"
aiofiles = "*"
aiologger = "^0.7.0"
pyyaml = "^6.0.2"
sqlalchemy-utils = "^0.41.2"
sqlalchemy = "^2.0.36"
alembic = "^1.13.3"
fastapi-limiter = "^0.1.6"
redis = "5.0.8"
redisearch = "2.0.0"
python-multipart = "*"
python-dotenv = "^1.0.1"
celery = "^5.4.0"
fastapi-cache2 = "^0.2.2"
aiocache = "^0.12.3"
dvc-s3 = "^3.2.0"
litserve = "^0.2.4"
gpustat = "^1.1.1"
nvitop = "^1.3.2"
gradio = "5.7.1"
gradio-client = "^1.5.0"
accelerate = "^1.1.1"
cryptography = "^44.0.0"
boto3 = "*"
pyopenssl = "^24.3.0"

[tool.poetry.dev-dependencies]
pytest-asyncio = "^0.20.3"

[[tool.poetry.source]]
name = "pytorch_cuda"
url = "https://download.pytorch.org/whl/cu124"
priority = "explicit"

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