-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
104 lines (90 loc) · 2.47 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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
[tool.poetry]
name = "scholarsense"
version = "0.1.0"
description = ""
authors = ["TERBOUCHE Hacene <[email protected]>"]
license = "MIT"
readme = "README.md"
packages = [ { include = "scholar_sense", from = "src" } ]
[tool.poetry.dependencies]
python = ">=3.9,<3.9.7 || >3.9.7,<3.12"
arxiv = "^1.4.8"
pandas = "^2.0.3"
tqdm = "^4.65.0"
nltk = "^3.8.1"
docarray = {extras = ["qdrant"], version = "^0.37.0"}
torch = { version="^1.13.0", extras=["cuda116"] }
torchvision = { version="^0.14", extras=["cuda116"] }
sentence-transformers = "^2.2.2"
streamlit = "^1.25.0"
openai = "^0.27.8"
typer = {extras = ["all"], version = "^0.9.0"}
[tool.poe.tasks]
force-cuda113 = "python -m pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html"
force-cuda116 = "python -m pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1+cu116 -f https://download.pytorch.org/whl/torch_stable.html"
[tool.poetry.group.dev.dependencies]
ipdb = "^0.13"
wandb = "^0.13.5"
matplotlib = "^3.6.2"
poethepoet = "^0.16.4"
ipywidgets = "^8.0.4"
jupyterlab = "^3.6.1"
seaborn = "^0.12.2"
wordcloud = "^1.9.2"
[tool.poetry.group.lint.dependencies]
black = {extras = ["jupyter"], version = "^23.7.0"}
flake8 = "^5.0"
isort = "^5.10"
mypy = ">=0.991"
pre-commit = "^2.16"
[tool.poetry.group.tests.dependencies]
pytest = "^7.1"
pytest-cov = "^3.0"
pytest-mock = "^3.8"
[tool.poetry.group.docs.dependencies]
Sphinx = ">=5.1"
myst-nb = "^0.16"
sphinx-autoapi = ">=1.8"
sphinx-rtd-theme = ">=1.0"
[tool.poetry.scripts]
scholarsense = "scholar_sense.main:app"
[tool.black]
target-version = ['py39']
line_length = 89
include = '\.pyi?$'
exclude = '''
(
/(
\.eggs # exclude a few common directories in the
| \.git # root of the project
| \.hg
| \.mypy_cache
| \.tox
| \.venv
| _build
| buck-out
| build
| dist
)/
)
'''
[tool.isort]
profile = "black"
line_length = 89
multi_line_output = 3
include_trailing_comma = true
force_grid_wrap = 0
use_parentheses = true
[tool.semantic_release]
version_toml = "pyproject.toml:tool.poetry.version"
# to track and update version in other files:
# version_variable = ["some/file:__version__", ...]
branch = "main"
changelog_file = "CHANGELOG.md"
build_command = "pip install poetry && poetry build"
dist_path = "dist/"
upload_to_pypi = false
remove_dist = false
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"