pyproject.toml

[project]
name = "pnp-diffusion-features"
version = "0.1.0"
description = ""
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
    "albumentations>=1.4.24",
    "clip",
    "diffusers==0.12.1",
    "einops>=0.8.0",
    "gradio>=5.12.0",
    "hf-transfer>=0.1.9",
    "imageio[ffmpeg]>=2.36.1",
    "kornia>=0.7.4",
    "omegaconf>=2.3.0",
    "opencv-python-headless>=4.10.0.84",
    "python-slugify>=8.0.4",
    "pytorch-lightning==1.9.0",
    "scikit-learn>=1.6.0",
    "taming-transformers",
    "torch==1.13.1",
    "torchvision>=0.14.1",
    "transformers>=4.48.0",
]

[tool.ruff]
line-length = 119

[tool.ruff.lint]
select = ["ALL"]
ignore = [
    "COM812", # missing-trailing-comma
    "D203",   # one-blank-line-before-class
    "D213",   # multi-line-summary-second-line
    "E501",   # line-too-long
    "SIM117", # multiple-with-statements
]
extend-ignore = [
    "D100",    # undocumented-public-module
    "D101",    # undocumented-public-class
    "D102",    # undocumented-public-method
    "D103",    # undocumented-public-function
    "D104",    # undocumented-public-package
    "D105",    # undocumented-magic-method
    "D107",    # undocumented-public-init
    "EM101",   # raw-string-in-exception
    "FBT001",  # boolean-type-hint-positional-argument
    "FBT002",  # boolean-default-value-positional-argument
    "PD901",   # pandas-df-variable-name
    "PGH003",  # blanket-type-ignore
    "PLR0913", # too-many-arguments
    "PLR0915", # too-many-statements
    "TRY003",  # raise-vanilla-args
]
unfixable = [
    "F401", # unused-import
]

[tool.ruff.format]
docstring-code-format = true

[tool.uv.sources]
clip = { git = "https://github.com/openai/CLIP.git" }
taming-transformers = { git = "https://github.com/CompVis/taming-transformers.git" }