inner-product-transforms/pyproject.toml at main · aidos-lab/inner-product-transforms · GitHub

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
[project]
name = "shape-synthesis"
version = "0.1.0"
description = "Shape synthesis"
authors = [{ name = "Your Name", email = "you@example.com" }]
requires-python = ">=3.10,<3.13"
readme = "README.md"
dependencies = [
    "lightning>=2.4.0,<3",
    "torch-geometric>=2.6.1,<3",
    "torchvision>=0.20.1,<0.21",
    "kaolin",
    "pydantic>=2.10.4,<3",
    "wandb>=0.19.1,<0.20",
    "matplotlib>=3.10.0,<4",
    "pyvista[jupyter]>=0.44.2,<0.45",
    "scikit-learn>=1.6.0,<2",
    "trame>=3.7.6,<4",
    "ipywidgets>=8.1.5,<9",
    "ipykernel>=6.29.5,<7",
    "torch-fidelity>=0.3.0",
    "gdown>=5.2.0",
    "torch>=2.5.1",
    "torchinfo>=1.8.0",
    "chamfer-3d",
    "emd-ext",
]


[dependency-groups]
dev = [
    "poethepoet>=0.25.0,<0.26",
    "ipykernel>=6.29.5,<7",
    "basedpyright>=1.31.4",
]
vis = ["pytest>=8.3.4,<9", "pandas>=2.2.3,<3"]


[tool.uv]
default-groups = ["dev", "vis"]

[[tool.uv.index]]
name = "kaolin-source"
url = "https://nvidia-kaolin.s3.us-east-2.amazonaws.com/index.html"
explicit = true

[tool.uv.sources]
kaolin = { url = "https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.5.1_cu124/kaolin-0.17.0-cp310-cp310-linux_x86_64.whl" }
chamfer-3d = { path = "src/dependencies/dist/chamfer_3d-0.0.0-cp310-cp310-linux_x86_64.whl" }
emd-ext = { path = "src/dependencies/dist/emd_ext-0.0.0-cp310-cp310-linux_x86_64.whl" }

[tool.pyright]
# strict = ["strec"]
ignore = ["tests"]
venvPath = "."
venv = ".venv"


[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
sources = ['src']

[project.scripts]
main = "shape_synthesis.main:main"
trainvae = "train_vae:main"