This repository has been archived by the owner on Jun 23, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpyproject.toml
62 lines (58 loc) · 2.8 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
[tool.poetry]
name = "oc"
version = "0.1.0"
description = ""
authors = ["Alan Blanchet <[email protected]>"]
readme = "README.md"
repository = "https://github.com/AlanBlanchet/SegmentezDesClientsDunSiteECommerce"
packages = [{ include = "src/" }]
[tool.poetry.dependencies]
python = "~3.10"
bs4 = "^0.0.1"
ipywidgets = "^8.0.6"
regex = "^2023.6.3"
scikit-learn = "^1.3.0"
matplotlib = "^3.7.1"
tqdm = "^4.65.0"
nltk = "^3.8.1"
torch = { url = "https://download.pytorch.org/whl/cu118/torch-2.0.1%2Bcu118-cp310-cp310-linux_x86_64.whl" }
# Rapids manual installation - Lead to bug with tensorflow
# -------------------------------------------------------------------
cuml-cu11 = { url = "https://pypi.nvidia.com/cuml-cu11/cuml_cu11-23.4.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
cudf-cu11 = { url = "https://pypi.nvidia.com/cudf-cu11/cudf_cu11-23.4.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
dask-cudf-cu11 = { url = "https://pypi.nvidia.com/dask-cudf-cu11/dask_cudf_cu11-23.4.1-py3-none-any.whl" }
raft-dask-cu11 = { url = "https://pypi.nvidia.com/raft-dask-cu11/raft_dask_cu11-23.4.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
cubinlinker-cu11 = { url = "https://pypi.nvidia.com/cubinlinker-cu11/cubinlinker_cu11-0.3.0.post1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
ptxcompiler-cu11 = { url = "https://pypi.nvidia.com/ptxcompiler-cu11/ptxcompiler_cu11-0.7.0.post1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
rmm-cu11 = { url = "https://pypi.nvidia.com/rmm-cu11/rmm_cu11-23.4.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
pylibraft-cu11 = { url = "https://pypi.nvidia.com/pylibraft-cu11/pylibraft_cu11-23.4.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
ucx-py-cu11 = { url = "https://pypi.nvidia.com/ucx-py-cu11/ucx_py_cu11-0.31.1-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl" }
python-utils = "^3.7.0"
# -------------------------------------------------------------------
tsne-torch = { git = "https://github.com/AlanBlanchet/tsne-pytorch" }
gensim = "^4.3.1"
transformers = { extras = ["torch"], version = "^4.31.0" }
pytorch-lightning = "^2.0.5"
# pyLDAvis didn't match with pandas version < 2
pyldavis = { git = "https://github.com/AlanBlanchet/pyLDAvis.git" }
datasets = "^2.14.1"
ipykernel = "^6.25.0"
peft = "^0.4.0"
mlflow = "^2.5.0"
torchvision = "^0.15.2"
scikit-multilearn = "^0.2.0"
dash = "^2.13.0"
dash-bootstrap-components = "^1.5.0"
tensorflow = "^2.13.0"
tensorflow-hub = "^0.14.0"
gradient-accumulator = "^0.5.2"
pandarallel = "^1.6.5"
fastapi = "^0.103.1"
uvicorn = {extras = ["standard"], version = "^0.23.2"}
accelerate = "^0.23.0"
[tool.poetry.group.dev.dependencies]
black = "^23.3.0"
isort = "^5.12.0"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"