CheckEmbed/pyproject.toml at main · spcl/CheckEmbed · 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
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "CheckEmbed"
version = "0.0.1"
authors = [
  { name="Maciej Besta", email="maciej.besta@inf.ethz.ch" },
  { name="Lorenzo Paleari", email="lpaleari@student.ethz.ch" },
  { name="Ales Kubicek", email="akubicek@student.ethz.ch" },
  { name="Robert Gerstenberger", email="gerstenberger.robert@gmail.com" },
  { name="Patrick Iff", email="patrick.iff@inf.ethz.ch" },
  { name="Eric Schreiber", email="eric.schreiber@inf.ethz.ch" },
]
description = "Python package for 'CheckEmbed'"
readme = "README.md"
license = {file = "LICENSE"}
requires-python = ">=3.8.12,<3.11"
classifiers = [
  "Programming Language :: Python :: 3",
  "Operating System :: OS Independent",
]
dependencies = [
  "accelerate>=0.30.1,<0.35.0",
  "backoff>=2.2.1,<3.0.0",
  "bert-score>=0.3.13,<1.0.0",
  "diffusers==0.33.1",
  "faker>=25.8.0,<26.0.0",
  "huggingface-hub>=0.30.0,<0.31.0",
  "joblib>=1.4.2,<2.0.0",
  "langchain>=0.2.11",
  "langchain_ollama",
  "matplotlib>=3.7.1,<4.0.0",
  "numpy>=1.24.3,<2.0.0",
  "openai>=1.0.0,<2.0.0",
  "packaging>=24.1,<25.0",
  "pandas>2.0.0,<3.0.0",
  "pillow>=11.2.1",
  "pydantic>=2.8.2",
  "scikit-learn>=1.5.1,<2.0.0",
  "scipy>1.15.0,<2.0.0",
  "seaborn>=0.13.2,<0.14.0",
  "selfcheckgpt>=0.1.7,<1.0.0",
  "sentencepiece>=0.2.0,<0.3.0",
  "tiktoken>=0.7.0,<1.0.0",
  "torch==2.6.0",
  "transformers>=4.51.3,<5.0.0",
  "transformers[torch]",
  "wheel>=0.43.0,<1.0.0",
  "vllm>=0.8.2,<1.0.0"
]

[project.optional-dependencies]
cuda = [
  "xformers>=0.0.27,<0.0.40",
  "flash-attn>=2.5.6,<3.0.0",
]

[project.urls]
Homepage = "https://github.com/spcl/CheckEmbed"

[project.scripts]