emo-classifier/pyproject.toml at main · stdiff/emo-classifier · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
[tool.poetry]
name = "emo-classifier"
version = "0"
description = "Just a demonstration of a PyTorch project with goemotions data set"
authors = ["Hironori Sakai <crescent.lab@gmail.com>"]
license = "MIT License"
include = ["emo_classifier/artifact/*", "emo_classifier/resources/*"]

## libraries for deployable package
[tool.poetry.dependencies]
python = ">=3.9,<3.10"
onnxruntime = "~1.9.0"
pandas = "1.3.5"
scikit-learn = "1.0.2"
scipy = "~1.7.0"
numpy = "1.21.0"
spacy = "3.2.1"
en-core-web-md = {url = "https://github.com/explosion/spacy-models/releases/download/en_core_web_md-3.2.0/en_core_web_md-3.2.0-py3-none-any.whl"}
torch = {version = "1.12.0", python = ">=3.9,<3.10", platform = "linux"}
torchtext = {version = "0.13.0", python = ">=3.9,<3.10", platform = "linux"}
pytorch-lightning = "1.7.7"
fastapi = "0.70.1"
uvicorn = {extras = ["standard"], version = "~0.15.0"}
torchmetrics = "0.7.2"

## libraries for the training pipeline and tests
[tool.poetry.group.dev.dependencies]
jupyterlab = ">=3.5"
jupyterlab-code-formatter = "~1.4"
black = {version = "23.3.0", extras = ["jupyter"]}
isort = "^5.9.1"
jupytext = "~1.11"
jupyterlab-git = "~0.30"
jupyterlab-spellchecker = "~0.6"
jupyterlab-templates = "0.4.0"
watermark = "^2.2.0"
altair = "4.2.0"
jsonschema = ">=3.0,<4.0"
pyarrow = "^6.0.1"
pytest = "^6.2.5"
nbconvert = "^6.3.0"
boto3 = "1.20.43"
sagemaker = "2.73.0"
docker = "5.0.3"
streamlit = "1.7.0"
ray = {version = "1.11.0", extras = ["default"]}
opencensus = "0.8.0"

[tool.black]
line-length = 120
target-version = ["py39"]

[tool.pytest.ini_options]
minversion = "6.0"
addopts = "-ra -q"
## Only unittest. Do not add test/integration.
testpaths = [
    "test/emo_classifier"
]

[tool.poetry.scripts]
ingestion = "script.ingestion:start"
train = "script.train:start"
server = "emo_classifier.server:start"

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