| [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.15" | |
| 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+cpu", source = "pytorch_cpu"} | |
| torchvision = {version = "^0.19.1+cpu", source = "pytorch_cpu"} | |
| 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" | |
| pyopenssl = "^23.0.0" | |
| cryptography = "^41.0.0" | |
| accelerate = "^1.1.1" | |
| gradio="5.7.1" | |
| [tool.poetry.dev-dependencies] | |
| pytest-asyncio = "^0.20.3" | |
| [[tool.poetry.source]] | |
| name = "pytorch_cpu" | |
| url = "https://download.pytorch.org/whl/cpu" | |
| priority = "explicit" | |
| [build-system] | |
| requires = ["poetry-core"] | |
| build-backend = "poetry.core.masonry.api" | |