-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
37 lines (32 loc) · 886 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
[tool.poetry]
name = "jafx"
version = "0.1.0"
description = "Composable deep learning with effectful JAX primitives"
authors = []
[tool.poetry.dependencies]
python = ">=3.10,<3.11"
cattrs = "^1.9.0"
ansicolors = "^1.1.8"
matplotlib = "^3.5.1"
dm-haiku = {version = "^0.0.6", optional = true}
flax = {version = "^0.5.0", optional = true}
moviepy = {version = "^1.0.3", optional = true}
tensorboard = {version = "^2.9.0", optional = true}
pendulum = {version = "^2.1.2", optional = true}
tensorflow = "^2.9.1"
[tool.poetry.dev-dependencies]
ipdb = "^0.13.9"
pytest = "^6.2.5"
black = "^22.3.0"
isort = "^5.9.3"
[tool.poetry.extras]
haiku = ["dm-haiku"]
flax = ["flax"]
logging = ["moviepy", "tensorboard", "pendulum"]
[tool.isort]
profile = "black"
multi_line_output = 3
skip = "__init__.py"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"