pyproject.toml

[tool.poetry]
name = "museum-img-searcher"
version = "0.1.0"
description = ""
authors = ["JKearnsl <pooolg@hotmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.11.8"
datasets = "^2.18.0"
torch = "^2.2.2"
nvidia-cublas-cu12 = {url = "https://download.pytorch.org/whl/cu121/nvidia_cublas_cu12-12.1.3.1-py3-none-manylinux1_x86_64.whl#sha256=ee53ccca76a6fc08fb9701aa95b6ceb242cdaab118c3bb152af4e579af792728"}
nvidia-cusparse-cu12 = {url = "https://download.pytorch.org/whl/nightly/cu121/nvidia_cusparse_cu12-12.1.0.106-py3-none-manylinux1_x86_64.whl"}

torchvision = "^0.17.2"
scikit-learn = "^1.4.2"
tqdm = "^4.66.2"
matplotlib = "^3.8.4"
seaborn = "^0.13.2"
xgboost = "^2.0.1"

pika = "^1.3.2"
PyYAML = "^6.0.1"
python-dotenv= "^1.0.0"
python-consul = "^1.1.0"
boto3 = "^1.34.84"
sqlalchemy = "^2.0.29"
pgvector = "^0.2.5"
psycopg2-binary = "^2.9.9"

[tool.poetry.group.dev.dependencies]
ipykernel = "^6.29.4"
pika-stubs = "^0.1.3"

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