vizion3D/pyproject.toml at 4a25b74e273c66f9746677a826f0d7287176cc3b · OlafenwaMoses/vizion3D · 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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "vizion3d"
version = "1.0.0"
description = "U3D Capability and Intelligence for the real world. "
readme = "README.md"
license = { text = "MIT" }
requires-python = ">=3.12"
keywords = [
    "3d", "computer-vision", "depth-estimation", "point-cloud",
    "mesh", "reconstruction", "nerf", "slam", "lidar", "ai",
    "artificial intelligence", "llm"
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: Image Processing",
    "Topic :: Multimedia :: Graphics :: 3D Rendering",
    "Typing :: Typed",
]
dependencies = [
    "fastapi>=0.115.0",
    "clean-ioc>=1.3.0",
    "pydantic>=2.0.0",
    "grpcio>=1.60.0",
    "grpcio-tools>=1.60.0",
    "uvicorn>=0.23.0",
    "python-multipart>=0.0.6",
    "transformers>=5.6.2",
    "pillow>=12.2.0",
    "open3d>=0.18.0",
    "orjson>=3.9.0",
    "ultralytics>=8.3.0",
]

# ---------------------------------------------------------------------------
# Hardware extras
#
# cpu / mps
#   Wheels are on PyPI — pip resolves them automatically.
#
#   pip install vizion3d[cpu]
#   pip install vizion3d[mps]
#
# cuda
#   CUDA wheels bundle their own NCCL and are NOT on PyPI.
#   Install the matching wheel BEFORE `pip install vizion3d[cuda]`.
#   Using pinned versions avoids NCCL symbol mismatches (e.g. ncclCommWindowDeregister).
#
#   CUDA 12.4 (recommended):
#     pip install torch==2.5.1+cu124 torchvision==0.20.1+cu124 \
#       --index-url https://download.pytorch.org/whl/cu124
#
#   CUDA 12.1:
#     pip install torch==2.5.1+cu121 torchvision==0.20.1+cu121 \
#       --index-url https://download.pytorch.org/whl/cu121
#
#   CUDA 11.8:
#     pip install torch==2.5.1+cu118 torchvision==0.20.1+cu118 \
#       --index-url https://download.pytorch.org/whl/cu118
#
# amd
#   ROCm wheels are also NOT on PyPI.  Install first:
#
#   ROCm 6.2 (recommended):
#     pip install torch==2.5.1+rocm6.2 torchvision==0.20.1+rocm6.2 \
#       --index-url https://download.pytorch.org/whl/rocm6.2
#
#   ROCm 6.1:
#     pip install torch==2.5.1+rocm6.1 torchvision==0.20.1+rocm6.1 \
#       --index-url https://download.pytorch.org/whl/rocm6.1
#
# Colab
#   torch + CUDA is pre-installed — do not use any extras:
#     pip install vizion3d --no-deps
#     pip install fastapi "clean-ioc>=1.3.0" "pydantic>=2.0.0" grpcio grpcio-tools \
#       uvicorn python-multipart "transformers>=5.6.2" "pillow>=12.2.0" "open3d>=0.18.0"
#
# If a matching torch is already present, pip will not downgrade it.
# ---------------------------------------------------------------------------
[project.optional-dependencies]
cpu  = ["torch>=2.4.0", "torchvision>=0.19.0"]
mps  = ["torch>=2.4.0", "torchvision>=0.19.0"]
cuda = []
amd  = []

[project.urls]
Homepage = "https://github.com/OlafenwaMoses/vizion3D"
Documentation = "https://olafenwamoses.github.io/vizion3D"
Repository = "https://github.com/OlafenwaMoses/vizion3D"
Issues = "https://github.com/OlafenwaMoses/vizion3D/issues"

[project.scripts]
vizion3d-serve-rest = "vizion3d.server.rest.app:run"
vizion3d-serve-grpc = "vizion3d.server.grpc.server:serve"

[dependency-groups]
dev = [
    "mkdocs>=1.6,<2.0",
    "mkdocs-material>=9.7.6,<10.0",
    "mkdocstrings[python]>=1.0.4",
    "pytest>=9.0.3",
    "ruff>=0.9.0",
    "twine>=6.0.0",
]

# ---------------------------------------------------------------------------
# Hatch build — wheel & sdist
# ---------------------------------------------------------------------------

[tool.hatch.build.targets.wheel]
packages = ["vizion3d"]

[tool.hatch.build.targets.sdist]
include = [
    "vizion3d/",
    "tests/",
    "docs/",
    "mkdocs.yml",
    "README.md",
    "pyproject.toml",
]
exclude = [
    "docs/assets/pointclouds/",
    "docs/assets/images/",
]

# ---------------------------------------------------------------------------
# Pytest
# ---------------------------------------------------------------------------

[tool.pytest.ini_options]
testpaths = ["tests"]
addopts   = "-v"

# ---------------------------------------------------------------------------
# Ruff
# ---------------------------------------------------------------------------

[tool.ruff]
line-length = 100
target-version = "py312"
extend-exclude = ["*_pb2*.py", "stereo_sample.py"]

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