File size: 962 Bytes
f7a65ec
 
 
 
 
 
 
 
 
 
 
 
585ddc0
 
9e4713f
 
 
 
 
743cb9b
f7a65ec
 
bd70023
 
 
 
 
 
9e4713f
 
 
f7a65ec
 
 
bd70023
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
[tool.poetry]
name = "shad-mlops-transformers"
version = "1.0.0"
description = "My first mlops project with transformers on huggingface"
authors = ["alekseik1 <[email protected]>"]
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"