[tool.poetry]
name = "shad-mlops-transformers"
version = "1.0.0"
description = "My first mlops project with transformers on huggingface"
authors = ["alekseik1 <1alekseik1@gmail.com>"]
license = "MIT"
readme = "README.md"
packages = [{include = "shad_mlops_transformers"}]

[tool.poetry.dependencies]
python = "^3.10"
streamlit = "^1.21.0"
torch = "^1.13"
transformers = "^4.27.4"
pydantic = "^1.10.7"
scikit-learn = "^1.2.2"
numpy = "^1.24.2"
loguru = "^0.7.0"
dvc = {version = "^2.54.0", extras = ["s3"]}
matplotlib = "^3.7.1"


[tool.poetry.group.dev.dependencies]
flake8 = "^6.0.0"
black = "^23.3.0"
docformatter = "^1.6.0"
isort = "^5.12.0"

[tool.poetry.scripts]
train = "shad_mlops_transformers.trainer:main"

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

[tool.black]
line-length = 120

[tool.isort]
line_length = 120
profile = "black"

[tool.docformatter]
wrap-summaries = 119
wrap-description = 110
black = "true"