pyproject.toml

[project]
name = "diffusion_policy"
version = "0.1.0"
description = "Diffusion Policy for Visuomotor Policy Learning"
readme = "README.md"
requires-python = ">=3.9,<3.12"
license = {text = "MIT"}
dependencies = [
    # Core ML
    "torch>=2.1.0",
    "torchvision>=0.16.0",

    # Numerical computing
    "numpy>=1.23.3,<2.0",
    "numba>=0.56.4",
    "scipy>=1.9.1",

    # Image/Video processing
    "opencv-python>=4.6.0",
    "scikit-image>=0.19.3",
    "scikit-video>=1.1.11",
    "imageio>=2.22.0",
    "imageio-ffmpeg>=0.4.7",
    "av>=10.0.0",

    # Config/Logging
    "hydra-core>=1.2.0",
    "omegaconf>=2.2.0",
    "wandb>=0.13.3",
    "tensorboard>=2.10.1",
    "tensorboardx>=2.5.1",

    # Data handling
    "zarr>=2.12.0,<3.0",
    "numcodecs>=0.10.2,<0.14",
    "h5py>=3.7.0",
    "datasets>=2.6.1",

    # RL/Robotics
    "gym>=0.23.0,<0.24.0",
    "pymunk>=6.2.1,<7.0",
    "dm-control>=1.0.9",
    "pygame>=2.1.2",
    "pybullet>=3.1.6",

    # Diffusion/Transformers
    "diffusers>=0.11.1",
    "accelerate>=0.13.2",
    "einops>=0.4.1",

    # Utilities
    "tqdm>=4.64.1",
    "dill>=0.3.5",
    "matplotlib>=3.6.1",
    "shapely>=1.8.4",
    "cython>=0.29.32",
    "termcolor>=2.0.1",
    "psutil>=5.9.2",
    "click>=8.0.4",
    "boto3>=1.24.96",
    "threadpoolctl>=3.1.0",
    "cffi>=1.15.1",

    # Ray for distributed training
    "ray[default,tune]>=2.2.0",

    # Robomimic (must be 0.2.0 - 0.3.0 removed CropRandomizer)
    "robomimic==0.2.0",

    # Robosuite
    "robosuite>=1.5.0",

    # Video models
    "pytorchvideo>=0.1.5",
]

[project.optional-dependencies]
dev = [
    "pytest",
    "ipykernel",
]
real = [
    # Real robot dependencies - may need system libraries
    "pyrealsense2>=2.51.1",
    "pynput>=1.7.6",
]

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[tool.setuptools.packages.find]
where = ["."]

[tool.uv]
# Use PyTorch index for CUDA 12.x wheels
extra-index-url = ["https://download.pytorch.org/whl/cu124"]