-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathpyproject.toml
84 lines (72 loc) · 2.08 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
[tool.poetry]
name = "ml-competition-template"
version = "0.1.0"
description = "table data ml competition template"
authors = ["hikaru yamashita <[email protected]>"]
readme = "README.md"
[tool.poetry.dependencies]
python = "^3.9"
[tool.poetry.group.dev.dependencies]
black = "^23.1.0"
isort = "^5.10.1"
mdformat-gfm = "^0.3.5"
mdformat-frontmatter = "^2.0.1"
mdformat_footnote = "^0.1.1"
flake8 = "^5.0.0"
mypy = "^1.0.1"
pytest-cov = "^4.0.0"
ipython = "^8.14.0"
jupyter = "^1.0.0"
cudf-cu11 = { version="^23.06.0", source="rapids" }
dask-cudf-cu11 = { version="^23.06", source="rapids" }
cuml-cu11 = { version="^23.06", source="rapids" }
ptxcompiler-cu11 = { version="^0.7", source="rapids" }
cubinlinker-cu11 = { version="^0.3", source="rapids" }
rmm-cu11 = { version="^23.06", source="rapids" }
raft-dask-cu11 = { version="^23.06", source="rapids" }
ucx-py-cu11 = { version="^0.32", source="rapids" }
pylibraft-cu11 = { version="^23.06", source="rapids" }
torch = { version="^2.0", source="torch_cu118" }
torchvision = { version="^0.15", source="torch_cu118" }
cupy-cuda12x = "^12.1.0"
omegaconf = "^2.3.0"
opencv-python = "^4.7.0.72"
pytorch-lightning = "^2.0.4"
segmentation-models-pytorch = "^0.3.3"
torch-optimizer = "^0.3.0"
pandarallel = "^1.6.5"
imbalanced-learn = "^0.10.1"
mlcrate = "^0.2.0"
seaborn = "^0.12.2"
matplotlib-venn = "^0.11.9"
kaggle = "^1.5.14"
gensim = "^4.3.1"
nltk = "^3.8.1"
transformers = "^4.30.2"
sentencepiece = "^0.1.99"
keras = "^2.13.1"
ktboost = "^0.2.2"
catboost = "^1.2"
rgf-python = "^3.12.0"
tsfresh = "^0.20.1"
darts = "^0.24.0"
nbdime = "^3.2.1"
jupyter-black = "^0.3.4"
nlplot = "^1.6.0"
xgboost = "^1.7.6"
polars = "^0.18.4"
xfeat = { git = "https://github.com/karunru/xfeat.git", branch = "master" }
[[tool.poetry.source]]
name = "PyPI"
priority = "primary"
[[tool.poetry.source]]
name = "rapids"
url = "https://pypi.nvidia.com"
priority = "explicit"
[[tool.poetry.source]]
name = "torch_cu118"
url = "https://download.pytorch.org/whl/cu118"
priority = "explicit"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"