-
Notifications
You must be signed in to change notification settings - Fork 2
/
pyproject.toml
73 lines (64 loc) · 1.66 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
[tool.poetry]
name = "learned-point-cloud-compression-for-classification"
version = "0.1.0"
description = ""
authors = ["Mateen Ulhaq <[email protected]>"]
license = "MIT"
readme = "README.md"
packages = [{include = "src"}]
# packages = [{include = "learned_point_cloud_compression_for_classification"}]
[tool.poetry.scripts]
compressai-train = "src.run.train:main"
compressai-eval = "src.run.eval_model:main"
compressai-plot = "src.run.plot_rd:main"
[tool.poetry.dependencies]
python = ">=3.8.0,<3.11"
accelerate = "0.15.0"
aim = "^3.16.0"
bjontegaard = "^1.2.0"
catalyst = "22.04"
einops = "^0.7.0"
h5py = "^3.0.0"
hydra-core = "^1.2"
matplotlib = "^3.4.3"
memory-tempfile = "^2.2.3"
numpy = "^1.21.2"
omegaconf = "^2.2.2"
pandas = "^1.4.3"
Pillow = "^9.0.0"
plotly = "^5.0.0"
plyfile = "^0.9"
pyntcloud = {git = "https://github.com/YodaEmbedding/pyntcloud.git", rev = "12ee9f2208f4207844be80ac5fdbafaf9f0652fa"}
pytest = "^7.0.0"
pytorch-msssim = "^0.2.1"
PyYAML = "^6.0.0"
scipy = "^1.6.0"
seaborn = "^0.12.0"
tensorboard = "^2.9.0"
toml = "^0.10.2"
torch = ">=1.8,<2.0"
torch-geometric = "^2.3.1"
torchvision = ">=0.9,<0.15"
tqdm = "^4.66.1"
typing-extensions = "^4.0.0"
[tool.poetry.dev-dependencies]
black = "^23.1.0"
isort = "^5.10.1"
[tool.poetry.group.docs]
optional = true
[tool.poetry.group.docs.dependencies]
sphinx = "^4.0"
sphinx-book-theme = "^1.0.0"
sphinx-serve = "^1.0.1"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.black]
line-length = 88
[tool.isort]
profile = "black"
skip_gitignore = true
known_third_party = "PIL,pytorch_msssim,torchvision,torch"
[tool.mypy]
allow_redefinition = true
ignore_missing_imports = true