summaryrefslogtreecommitdiff
path: root/pyproject.toml
blob: e791dd93c602bcc1129f98aa1ac3ab4eb98ca788 (plain)
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
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
[tool.poetry]
name = "text-recognizer"
version = "0.1.0"
authors = ["aktersnurra <gustaf.rydholm@gmail.com>"]
description = "A text recognizer using best pratices in python and deep learning."
license = "MIT"
readme = "README.md"
homepage = "https://github.com/aktersnurra/text-recognizer"
repository = "https://github.com/aktersnurra/text-recognizer"
keywords = ["text recognizer, deep learning, pytorch"]

[tool.poetry.dependencies]
python = "^3.8"
click = "^7.1.2"
flake8-annotations = "^2.1.0"
flake8-docstrings = "^1.5.0"
darglint = "^1.3.1"
marshmallow = "^3.6.0"
sphinx-autodoc-typehints = "^1.10.3"
sphinx_rtd_theme = "^0.4.3"
boltons = "^20.1.0"
h5py = "^3.2.1"
toml = "^0.10.1"
torch = "^1.8.0"
torchvision = "^0.9.0"
loguru = "^0.5.0"
matplotlib = "^3.2.1"
tqdm = "^4.46.1"
pytest = "^5.4.3"
opencv-python = "^4.3.0"
nltk = "^3.5"
torch-summary = "^1.4.2"
defusedxml = "^0.6.0"
omegaconf = "^2.0.2"
wandb = "^0.10.12"
einops = "^0.3.0"
gtn = "^0.0.0"
sentencepiece = "^0.1.95"
pytorch-lightning = "^1.2.4"
Pillow = "^8.1.2"

[tool.poetry.dev-dependencies]
pytest = "^5.4.2"
coverage = {extras = ["toml"], version = "^5.1"}
pytest-cov = "^2.9.0"
pytest-mock = "^3.1.0"
black = "^19.10b0"
flake8 = "^3.8.2"
flake8-bandit = "^2.1.2"
flake8-black = "^0.2.0"
flake8-bugbear = "^20.1.4"
flake8-import-order = "^0.18.1"
safety = "^1.9.0"
mypy = "^0.770"
pytype = {version = "^2020.5.13", python = "3.7"}
typeguard = "^2.7.1"
xdoctest = "^0.12.0"
sphinx = "^3.0.4"
jupyter = "^1.0.0"
gpustat = "^0.6.0"
redlock-py = "^1.0.8"
wandb = "^0.10.11"
graphviz = "^0.16"
scipy = "^1.6.1"

[tool.coverage.report]
fail_under = 50

[tool.poetry.scripts]
download-emnist = "text_recognizer.data.emnist:download_emnist"
download-nltk = "text_recognizer.data.sentence_generator:load_nltk_brown_corpus"
generate-emnist-lines = "text_recognizer.data.emnist_lines:generate_emnist_lines"
download-iam = "text_recognizer.data.iam:download_iam"
generate-iam-lines = "text_recognizer.data.iam_lines:generate_iam_lines"
extract-iam-text = "text_recognizer.data.iam_preprocessor:cli"
make-wordpieces = "text_recognizer.data.make_wordpieces:cli"
build-transitions = "text_recognizer.data.build_transitions:cli"

create-emnist-support-files = "text_recognizer.tests.support.create_emnist_support_files:create_emnist_support_files"
create-emnist-lines-datasets = "text_recognizer.datasets.emnist_lines_dataset:create_datasets"
create-iam-paragraphs = "text_recognizer.datasets.iam_paragraphs_dataset:main"
prepare-experiments = "training.prepare_experiments:run_cli"
run-experiment = "training.run_experiment:run_cli"


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