capo/pyproject.toml at main · finitearth/capo · 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
[tool.poetry]
name = "capo"
version = "0.1.0"
description = "Cost-aware prompt tuning algorithm"
authors = ["Anonymous"]
readme = "README.md"

[tool.poetry.dependencies]
python = "~3.12"
promptolution = "^1.3.2"
numpy = "~1.26"
pandas = "^2.2.3"
scipy = "^1.15.2"
pyarrow = "^19.0.1"
fastparquet = "^2024.11.0"
huggingface-hub = "^0.29.2"
datasets = "^3.3.2"
torch = "^2.5.1"
promptwizard = {git = "https://github.com/HappyAnonymousQuokka/promptwizardIO"}

[tool.poetry.group.visualization.dependencies]
matplotlib = "^3.10.1"
ipykernel = "^6.29.5"
seaborn = "^0.13.2"
plotly = "^6.0.1"
nbformat = "^5.10.4"

[tool.poetry.group.dev.dependencies]
pre-commit = "^4.1.0"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.black]
line-length = 100

[tool.isort]
line_length = 100
profile = "black"