NvidiaOpticalFlowSDKforPytorch/pyproject.toml at main · JuanFMontesinos/NvidiaOpticalFlowSDKforPytorch · 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
[project]
name = "torch-nvidia-of-sdk"
version = "5.0.2" # Matches that of Nvidia SDK
description = "PyTorch bindings for NVIDIA Optical Flow SDK, providing hardware-accelerated optical flow computation with PyTorch end-to-end integration in Nvidia and Python."
readme = "README.md"
authors = [
    { name = "Juan Montesinos", email = "jfmontgar@gmail.com" }
]
requires-python = ">=3.13"
dependencies = [
    "imageio[ffmpeg]>=2.37.2",
    "matplotlib>=3.10.8",
    "numpy",
    "torch>=2.10.0",
    "tqdm>=4.67.2",
]

[project.optional-dependencies]
full = [
    "opencv-contrib-python-headless>=4.13.0.90",
]

[build-system]
requires = [
    "scikit-build-core",
    "pybind11",
    "torch>=2.10.0",
    "numpy",
    "nvidia-cuda-runtime-cu12",
    "nvidia-cudnn-cu12"
]
build-backend = "scikit_build_core.build"

[tool.scikit-build]
wheel.packages = ["src/of"]
wheel.py-api = "py3"
build.verbose = true

[tool.ruff]
line-length = 120

[tool.ruff.lint.per-file-ignores]
"__init__.py" = ["F401"]  # Allow unused imports in __init__.py files

[dependency-groups]
dev = [
    "pytest>=7.0.0",
    "ruff>=0.15.0",
]