[tool.poetry]
name = "polos"
version = "0.1.3"
authors = ["Yuiga Wada <yuigawada@gmail.com>"]
description = "[CVPR24] Polos: Multimodal Metric Learning from Human Feedback for Image Captioning"
license = "BSD-3-Clause-Clear"
readme = "README.md"
classifiers = [
    'Development Status :: 4 - Beta',
    'Environment :: Console',
    'Intended Audience :: Science/Research',
    'Topic :: Scientific/Engineering :: Artificial Intelligence',
]
packages = [
    {include = "polos"},
]
include = [
    "LICENSE",
    "pyproject.toml",
    "CONTRIBUTING.md"
]

[tool.poetry.scripts]
polos = 'polos.cli:polos'

[tool.poetry.dependencies]
python = "3.9.2"
PyYAML = "5.3.*"
#numpy = "<=1.19"
protobuf="3.20.0"
pandas = "^1.0.0"
#fairseq = "0.9.0"
transformers = "4.*"
pytorch-lightning = "<=1.3"
pytorch-nlp = "0.5.0"
fsspec = "0.8.7"
psycopg2-binary = "^2.9.6"
torch = "^2.1.0"
click = "^8.1.7"
ftfy = "^6.1.1"
regex = "^2023.10.3"
wandb = "^0.15.12"
scikit-learn = "^1.3.2"
fairseq = "^0.12.2"
matplotlib = "^3.8.1"
pycocotools = "^2.0.7"
pycocoevalcap = "^1.2"
termcolor = "^2.3.0"
torchvision = "^0.16.0"
streamlit = "^1.35.0"
setuptools = "69.5.1"

[tool.poetry.dev-dependencies]
sphinx-markdown-tables = "0.0.15"
coverage = "^5.5"

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

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