shad-mlops-transformers / pyproject.toml
alekeik1's picture
feat(main): init train and dvc
9e4713f
raw
history blame
940 Bytes
[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"]}
[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"