mteb

1
[build-system]
2
requires = ["setuptools>=42", "wheel"]
3
build-backend = "setuptools.build_meta"
4

5
[project]
6
name = "mteb"
7
version = "1.7.58"
8
description = "Massive Text Embedding Benchmark"
9
readme = "README.md"
10
authors = [
11
    { name = "MTEB Contributors", email = "niklas@huggingface.co" },
12
    { email = "nouamane@huggingface.co" },
13
    { email = "info@nils-reimers.de" },
14
]
15
license = { file = "LICENSE" }
16
keywords = ["deep learning", "text embeddings", "benchmark"]
17
classifiers = [
18
    "Development Status :: 4 - Beta",
19
    "Environment :: Console",
20
    "Intended Audience :: Developers",
21
    "Intended Audience :: Information Technology",
22
    "License :: OSI Approved :: Apache Software License",
23
    "Operating System :: OS Independent",
24
    "Programming Language :: Python",
25
]
26
requires-python = ">=3.8"
27
dependencies = [
28
    "datasets>=2.19.0",
29
    "jsonlines",
30
    "numpy",
31
    "requests>=2.26.0",
32
    "scikit_learn>=1.0.2",
33
    "scipy",
34
    "sentence_transformers>=2.2.0",
35
    "torch",
36
    "tqdm",
37
    "rich",
38
    "pytrec-eval-terrier>=0.5.6",
39
    "pydantic>=2.0.0",
40
    "typing_extensions",
41
    "eval_type_backport",
42
    "polars>=0.20.22",
43
]
44

45

46
[project.urls]
47
homepage = "https://github.com/embeddings-benchmark/mteb"
48
"Huggingface Organization" = "https://huggingface.co/mteb"
49
"Source Code" = "https://github.com/embeddings-benchmark/mteb"
50

51
[project.scripts]
52
mteb = "mteb.cmd:main"
53

54
[project.optional-dependencies]
55
dev = ["ruff>=0.0.254", "pytest", "pytest-xdist"]
56

57

58
[tool.setuptools.packages.find]
59
exclude = ["tests", "results"]
60

61
[tool.setuptools.package-data]
62
"*" = ["*.json"]
63

64
[tool.ruff]
65
target-version = "py38"
66

67
[tool.ruff.lint]
68
select = ["F", "I", "E", "D"]
69
ignore = ["E501",   # line too long 
70
        "E741",     # ambiguous variable name
71
        "F403",     # undefined import
72
        "D100",     # Missing docstring in public module
73
        "D101",     # Missing docstring in public class
74
        "D102",     # Missing docstring in public method
75
        "D103",     # Missing docstring in public function
76
        "D104",     # Missing docstring in public package
77
        "D107",     # Missing docstring in __init__
78
        "D205",     # 1 blank line required between summary line and description
79
        "D415",     # First line should end with a period
80
]
81
ignore-init-module-imports = true
82

83
[tool.ruff.lint.pydocstyle]
84
convention = "google"
85

86
[tool.ruff.lint.flake8-annotations]
87
mypy-init-return = true
88
suppress-none-returning = true
89

90
[tool.semantic_release]
91
branch = "main"
92
version_toml = ["pyproject.toml:project.version"]
93
build_command = "python -m pip install build; python -m build"
94
tag_format = "{version}"
95

96
[tool.semantic_release.commit_parser_options]
97
major_types = ["breaking"]
98
minor_types = ["feat"]
99
patch_types = ["fix", "perf"]
100