-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathpyproject.toml
50 lines (35 loc) · 963 Bytes
/
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
[tool.poetry]
name = "ml-training"
version = "0.1.0"
description = ""
authors = ["Masato Naka <[email protected]>"]
readme = "README.md"
packages = []
[tool.poetry.dependencies]
python = ">=3.11,<3.13"
[tool.poetry.group.dev.dependencies]
ruff = "^0.9.0"
[tool.poetry.group.ray.dependencies]
pyarrow = "19.0.0"
pandas = "2.2.3"
ray = {extras = ["train"], version = "^2.10.0"}
[tool.poetry.group.river.dependencies]
river = "0.22.0"
flask = "2.3.3"
[tool.poetry.group.tensorflow.dependencies]
portpicker = "1.6.0"
tensorflow = "2.18.0"
tf-keras = "^2.17.0"
tensorflow-datasets = "^4.9.6"
tensorflow-recommenders = "0.7.2"
[tool.poetry.group.pytorch.dependencies]
torchvision = "^0.20.0"
torch = "^2.2.2"
[tool.poetry.group.streamlit.dependencies]
streamlit = "^1.39.0"
google-cloud-aiplatform = "^1.70.0"
google-auth = "^2.35.0"
google-cloud-run = "^0.10.9"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"