-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathpyproject.toml
91 lines (80 loc) · 2.76 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
[tool.poetry]
name = "jammy"
version = "0.0.1" # This should be overriden by the CI at publish time
description = "A Versatile ToolBox"
authors = ["Qin <[email protected]>"]
license = "MIT"
packages = [
{ include="jammy" },
{ include="jamtorch" },
{ include="jamnp" },
{ include="jamviz" },
{ include="jamweb" }
]
readme = "README.md"
documentation = "https://jammy.readthedocs.io/en/latest/"
[tool.poetry.dependencies]
python = ">=3.7,<3.10"
numpy = "^1.20.3"
scipy = "^1.6.3"
loguru = "^0.5.3"
einops = "^0.3.0"
hydra-core = "^1.1.0"
GitPython = "^3.1.17"
ipdb = "^0.13.8"
matplotlib = "^3.3.0"
gpustat = "^0.6.0"
tqdm = "^4.11.0"
attrs = "^20.3.0"
pyzmq = { version = "^22.3.0" }
# learn
wandb = { version = "^0.10.31", optional = true }
# torch
pytorch-lightning = { version = "^1.5.10", optional = true }
torch = { version = "^1.10.0", optional = true }
torchvision = { version = "^0.11.0", optional = true }
torchinfo = { version = "^1.5.3", optional = true }
# webs
tornado = { version = "^6.1", optional = true }
# storage
h5py = { version = "^3.4.0", optional = true }
msgpack = { version = "^1.0.2", optional = true }
msgpack-numpy = { version = "^0.4.7", optional = true }
pyarrow = { version = "^6.0.0", optional = true }
lmdb = { version = "^1.2.1", optional = true }
python-memcached = { version = "^1.59", optional = true }
# pro
line-profiler = { version = "^3.3.1", optional = true }
ubelt = { version = "^0.10.2", optional = true }
parse = { version = "^1.19.0", optional = true }
pudb = { version = "^2022.1", optional = true }
timeout-decorator = "^0.5.0"
filelock = "^3.8.0"
[tool.poetry.extras]
torch = ["wandb", "pytorch-lightning", "torch", "torchvision", "torchinfo"]
web = ["tornado"]
storage = ["h5py", "msgpack", "msgpack-numpy", "pyarrow", "lmdb", "python-memcached"]
pro = ["line-profiler", "ubelt", "parse", "pudb"]
all = ["wandb", "pytorch-lightning", "torch", "torchvision", "torchinfo", "tornado", "h5py", "msgpack", "msgpack-numpy", "pyarrow", "lmdb", "python-memcached", "line-profiler", "ubelt", "parse", "pudb"]
[tool.poetry.dev-dependencies]
pre-commit = "^2.13.0"
isort = "^5.8.0"
black = "^21.5b1"
pylint = "^2.8.3"
Sphinx = "^4.2.0"
Pallets-Sphinx-Themes = "^2.0.1"
sphinx-tabs = "^3.2.0"
m2r2 = "^0.3.1"
[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"
[tool.poetry.scripts]
jgpus="jammy.cli_scripts.gpu_sc:start_sever"
jgpuc="jammy.cli_scripts.gpu_sc:start_client"
jshs="jammy.cli_scripts.shell_executor:start_sever"
jshc="jammy.cli_scripts.shell_executor:echo_hello"
jshinfo="jammy.cli_scripts.shell_executor:echo_state"
jshka="jammy.cli_scripts.shell_executor:client_kill_all"
jinspect-file="jammy.cli_scripts.inspect_file:simple"
jsys-info="jammy.cli_scripts.sys_info:main"
jdk="jammy.cli_scripts.dk:my_app"