-
Notifications
You must be signed in to change notification settings - Fork 2
/
pyproject.toml
69 lines (63 loc) · 1.45 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
[tool.poetry]
name = "autotm"
version = "0.2.3.4"
description = "Automatic hyperparameters tuning for topic models (ARTM approach) using evolutionary algorithms"
authors = [
"Khodorchenko Maria <mariyaxod@yandex.ru>",
"Nikolay Butakov alipoov.nb@gmail.com"
]
readme = "README.md"
license = "Apache-2.0"
homepage = "https://autotm.readthedocs.io/en/latest/"
repository = "https://github.com/ngc436/AutoTM"
packages = [
{include = "autotm"}
]
[tool.poetry.dependencies]
python = ">=3.9, <3.12"
bigartm = "0.9.2"
protobuf = "<=3.20.0"
tqdm = "4.50.2"
numpy = "*"
PyYAML = "5.3.1"
dataclasses-json = "*"
mlflow = "*"
click = "8.0.1"
scipy = "<=1.10.1"
hyperopt = "*"
pymystem3 = "*"
nltk = "*"
plotly = "*"
spacy = ">=3.5"
spacy-langdetect = "*"
gensim = "^4.1.2"
pandas = "*"
billiard = "*"
dill = "*"
pytest = "*"
celery = ">=4.4.7"
redis = "3.5.3"
jinja2 = "3.0"
PyMySQL = "*"
psycopg2-binary = "*"
pymongo = "3.11.3"
scikit-learn = "^1.1.1"
#pydantic = "1.10.8"
pydantic = "2.6.0"
openai = "^1.31.0"
seaborn = "^0.13.2"
[tool.poetry.dev-dependencies]
black = "*"
sphinx = "*"
flake8 = "*"
[tool.pytest.ini_options]
log_cli = true
log_cli_level = "DEBUG"
log_cli_format = "%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)"
log_cli_date_format = "%Y-%m-%d %H:%M:%S"
[tool.poetry.scripts]
autotmctl = 'autotm.main:cli'
fitness-worker = 'autotm.main_fitness_worker:main'
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"