forked from wangyuxinwhy/uniem
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
42 lines (37 loc) · 812 Bytes
/
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
[tool.poetry]
name = "uniem"
version = "0.3.2"
description = "unified embedding model"
authors = ["wangyuxin <[email protected]>"]
license = "MIT"
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.8"
transformers = "^4.28.0"
accelerate = "^0.21.0"
tqdm = "^4.65.0"
typer = {extras = ["all"], version = "^0.9.0"}
datasets = "^2.12.0"
[tool.poetry.group.dev.dependencies]
pytest = "^7.3.1"
blue = "^0.9.1"
ruff = "^0.0.265"
isort = "^5.12.0"
pyright = "^1.1.314"
[tool.poetry.group.mteb.dependencies]
mteb = "^1.0.2"
openai = "^0.27.8"
sentence-transformers = "^2.2.2"
text2vec = "^1.2.1"
[tool.ruff]
line-length = 128
select = [
'E',
'F',
'I',
'N'
]
src = ['uniem', 'mteb-zh/mteb_zh', 'tests']
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"