-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
96 lines (91 loc) · 2.72 KB
/
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
[tool.poetry]
name = "rag-for-math-qa"
version = "0.2.0"
description = "Experimentation/analysis code and scripts"
authors = [
"Zachary Levonian <[email protected]>"
]
license = "MIT"
readme = "README.md"
packages = [{include = "experiment", from = "src"}, {include = "rag", from = "src"}]
repository = "https://github.com/DigitalHarborFoundation/rag-for-math-qa.git"
[tool.poetry.dependencies]
python = ">=3.10,<3.12"
poetry = "1.6.1"
streamlit = "^1.23.1"
pandas = "^2.0.2"
scipy = "^1.11.0"
argon2-cffi = "^21.3.0"
sqlalchemy = "^2.0.20"
psycopg2-binary = "^2.9.7"
numpy = "^1.24.3"
spacy = "^3.6.1"
statsmodels = "^0.14.0"
tabulate = "^0.9.0"
scikit-learn = "^1.3.0"
openai = "^0.28.0"
tiktoken = "^0.4.0"
python-dotenv = "^1.0.0"
evaluate = "^0.4.0"
bert-score = "^0.3.13"
bleurt = {git = "https://github.com/google-research/bleurt.git"}
tensorflow = {version = "^2.13.0" }
tensorflow-macos = { version = "^2.13.0", platform = "darwin", markers = "platform_machine=='arm64'" }
tensorflow-intel = { version = "^2.13.0", platform = "win32" }
tensorflow-cpu = [
{ version = "^2.13.0", platform = "linux", markers = "platform_machine!='arm64' and platform_machine!='aarch64'" },
{ version = "^2.13.0", platform = "darwin", markers = "platform_machine!='arm64' and platform_machine!='aarch64'" },]
tensorflow-cpu-aws = { version = "^2.13.0", platform = "linux", markers = "platform_machine=='arm64' or platform_machine=='aarch64'" }
# https://github.com/tensorflow/tensorflow/blob/adb39b04e9cb116df4659a7e2de9eea27e62f25c/tensorflow/tools/pip_package/setup.py#L107-L108
# https://github.com/python-poetry/poetry/issues/8271#issuecomment-1697740447
tensorflow-io-gcs-filesystem = [
{ version = ">= 0.23.1", markers = "platform_machine!='arm64' or platform_system!='Darwin'" },
{ version = "< 0.32.0", markers = "platform_system == 'Windows'" }
]
datasets = "2.10.0"
krippendorff = "^0.6.0"
skrub = {git = "https://github.com/skrub-data/skrub.git"}
[tool.poetry.group.dev.dependencies]
jupyter = "^1.0.0"
matplotlib = "^3.7.1"
black = "^22.12.0"
isort = "^5.12"
flake8 = "^6.0.0"
nbqa = "^1.6.0"
pre-commit = "^2.21.0"
pytest = "^7.2.1"
pytest-cov = "^4.0.0"
jupyterlab = "^4.0.2"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.black]
line-length = 120
include = '\.pyi?$'
exclude = '''
/(
.eggs # exclude a few common directories in the
| .git # root of the project
| .github
| .gitignore
| .hg
| .mypy_cache
| .tox
| .venv
| venv
| _build
| buck-out
| build
| ci
| data
| dist
| docs
| docsrc
)/
'''
[tool.isort]
profile = "black"
line_length = 79
multi_line_output = 3
include_trailing_comma = true
virtual_env = "venv"