-
Notifications
You must be signed in to change notification settings - Fork 9
/
pyproject.toml
87 lines (72 loc) · 2.23 KB
/
pyproject.toml
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
[tool.poetry]
name = "ml-mipt"
version = "1.0.0"
description = "Machine learning course at MIPT"
authors = ["Vladislav Goncharenko <[email protected]>, Radoslav Neychev <[email protected]>"]
license = "MIT License"
[tool.poetry.dependencies]
python = "^3.8"
scikit-learn = "^0.24.1"
matplotlib = "^3.3.4"
pandas = "^1.2.2"
numpy = "^1.20.1"
scipy = "^1.6.0"
statsmodels = "^0.12.2"
seaborn = "^0.11.1"
xgboost = "^1.3.3"
opencv-python = "^4.5.1"
torch = "^1.7.1"
torchvision = "^0.8.2"
torchsummary = "^1.5.1"
# basic
Pillow = {version = "^7.2.0", optional = true} # TODO: remove
tqdm = {version = "^4.56.2", optional = true} # TODO: remove
scikit-image = {version = "^0.18.1", optional = true} # TODO: remove week0_12 imread and resize
h5py = {version = "^3.1.0", optional = true} # parse cats and dogs dataset, maybe remove?
pydotplus = {version = "^2.0.2", optional = true} # graph visualization
eli5 = {version = "^0.11.0", optional = true} # week0_07 feature importance
PDPbox = {version = "^0.2.0", optional = true} # week0_07 feature importance
shap = {version = "^0.38.1", optional = true} # week0_07 feature importance
# advanced
ipywidgets = "^7.6.3" # week1_15 downloading mnist via torchvision
# nlp
nltk = "^3.5"
gensim = "^3.8.3"
spacy = "^3.1.1"
subword-nmt = "^0.3.7"
pytorch-transformers = "^1.2.0"
torchtext = "^0.8"
bokeh = "^2.3.0"
# rl
gym = {version = "^0.18.0", optional = true}
graphviz = "^0.16"
[tool.poetry.extras]
basic = ["Pillow", "tqdm", "scikit-image", "h5py", "pydotplus", "eli5", "PDPbox", "shap"]
nlp = ["nltk", "gensim", "spacy", "subword-nmt", "pytorch-transformers", "torchtext", "bokeh"]
rl = ["gym", "graphviz"]
[tool.poetry.dev-dependencies]
pre-commit = "^2.10.1"
ipykernel = "^5.4.3"
[tool.black]
line-length = 100
target-version = ["py38"]
[tool.isort]
multi_line_output = 3
include_trailing_comma = true
force_grid_wrap = 0
use_parentheses = true
ensure_newline_before_comments = true
line_length = 100
lines_after_imports = 2
[tool.nbqa.config]
black = "pyproject.toml"
isort = "pyproject.toml"
flake8 = "setup.cfg"
[tool.nbqa.addopts]
flake8 = ["--extend-ignore=E402"]
[tool.nbqa.mutate]
black = 1
isort = 1
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"