InfiniCube/pyproject.toml at main · nv-tlabs/InfiniCube · 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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
[build-system]
requires = ["setuptools>=45", "wheel"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
include = ["infinicube*"]

[project]
name = "infinicube"
version = "0.1.0"
requires-python = ">=3.10"
# Common dependencies shared by both environments
dependencies = [
    "tqdm",
    "numpy<2.0.0",
    "scipy",
    "webdataset",
    "line-profiler",
    "imageio[ffmpeg]",
    "opencv_python_headless",
    "decord",
]

# Console scripts entry points
[project.scripts]
vis-grid = "infinicube.visualize.visualize_grid:main"
vis-wds = "infinicube.visualize.webdataset_data:main"
vis-compare = "infinicube.visualize.visualize_grid_compare:main"
vis-dynamic = "infinicube.visualize.visualize_dynamic_gaussians:main"

# Optional dependency groups: completely separate main and waymo environments
[project.optional-dependencies]
# Main environment: PyTorch ecosystem and all core development dependencies
main = [
    "torch==2.2.0",
    "torchvision==0.17.0",
    "pytorch-lightning==1.9.4",
    "torch-scatter",
    "tensorboard",
    "matplotlib",
    "pandas",
    "rich",
    "pytest-benchmark",
    "omegaconf",
    "scikit-image",
    "flatten-dict",
    "scikit-learn",
    "wandb==0.16.6",
    "transformers",
    "ffmpeg-python",
    "linkify-it-py",
    "python-pycg",
    "point_cloud_utils==0.29.5",
    "loguru",
    "randomname",
    "einops",
    "nvidia-ml-py",
    "polyscope",
    "trimesh",
    "icecream",
    "boto3",
    "viser",
    "mediapy",
    "scikit-spatial",
    "ftfy",
    "openmim",
    "timm",
    "termcolor",
    "gsplat @ https://github.com/nerfstudio-project/gsplat/releases/download/v1.4.0/gsplat-1.4.0%2Bpt22cu121-cp310-cp310-linux_x86_64.whl",
    "fvdb @ https://research.nvidia.com/labs/toronto-ai/infinicube/wheels/fvdb-0.2.0+pt22cu121-cp310-cp310-linux_x86_64.whl",
    "diffsynth @ git+https://github.com/yifanlu0227/DiffSynth-Studio-InfiniCube",
    "open3d-pycg-cpu",
    "x-unet",
    'lpips',
    "qwen-vl-utils"
]

# Waymo environment: only waymo-related dependencies and necessary utilities
waymo = [
    "waymo-open-dataset-tf-2-11-0==1.6.1",
    "gdown",
    "pyquaternion",
]

# Development environment: code formatting and linting tools
dev = [
    "ruff",
]


# UV configuration for package sources and links
[tool.uv]
find-links = [
    "https://download.pytorch.org/whl/cu121",
    "https://data.pyg.org/whl/torch-2.2.0+cu121.html",
]


# ruff configuration
[tool.ruff.lint]
select = ["E", "F", "I"]

fixable = ["ALL"]