semantic-search-eval/pyproject.toml at main · machinelearningZH/semantic-search-eval · GitHub

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
[project]
name = "semsearcheval"
version = "1.1.0"
description = "A package for evaluating search-based models using various metrics."
authors = [
  { name = "Statistisches Amt Kanton Zürich, Team Data", email = "datashop@statistik.zh.ch" },
]
license = { file = "LICENSE" }
readme = "README.md"
requires-python = ">=3.13"
dependencies = [
    "accelerate>=1.13.0",
    "einops>=0.8.2",
    "de-core-news-sm",
    "openai>=1.77.0",
    "ordered-set>=4.1.0",
    "polars>=1.40.1",
    "pytest-mock>=3.15.1",
    "rank-bm25>=0.2.2",
    "ruamel-yaml>=0.19.1",
    "seaborn>=0.13.2",
    "sentence-transformers<=5.4.1",
    "transformers<5.0.0",
    "spacy>=3.8.14",
    "tiktoken>=0.12.0",
    "dotenv>=0.9.9",
    "peft>=0.19.1",
]

[dependency-groups]
dev = [
    "pre-commit>=4.2.0",
    "pytest>=8.3.5",
    "pytest-clarity>=1.0.1",
    "ruff>=0.11.8",
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
      packages = ["semsearcheval"]

[tool.uv.sources]
de-core-news-sm = { url = "https://github.com/explosion/spacy-models/releases/download/de_core_news_sm-3.8.0/de_core_news_sm-3.8.0-py3-none-any.whl" }

[tool.ruff]
line-length = 100
indent-width = 4

[tool.ruff.lint]
select = ["E", "F", "I001", "I002"]
ignore = ["E501"]
preview = true

[tool.ruff.format]
quote-style = "double"
indent-style = "space"
skip-magic-trailing-comma = false
line-ending = "auto"

[tool.ruff.lint.isort]
lines-after-imports = 2