90 lines
2.3 KiB
TOML
90 lines
2.3 KiB
TOML
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[project]
|
|
name = "lilith-content-understanding"
|
|
version = "0.1.0"
|
|
description = "Comprehensive image understanding for ML services"
|
|
requires-python = ">=3.10"
|
|
license = "MIT"
|
|
authors = [{ name = "transquinnftw", email = "quinn@ftw.codes" }]
|
|
keywords = ["ml", "image", "detection", "nsfw", "depth", "color", "composition", "scene", "understanding"]
|
|
classifiers = [
|
|
"Development Status :: 3 - Alpha",
|
|
"Intended Audience :: Developers",
|
|
"License :: OSI Approved :: MIT License",
|
|
"Programming Language :: Python :: 3",
|
|
"Programming Language :: Python :: 3.10",
|
|
"Programming Language :: Python :: 3.11",
|
|
"Programming Language :: Python :: 3.12",
|
|
"Programming Language :: Python :: 3.13",
|
|
]
|
|
|
|
dependencies = [
|
|
"torch>=2.0.0",
|
|
"pillow>=10.0.0",
|
|
"transformers>=4.30.0",
|
|
"pydantic>=2.0.0",
|
|
"numpy>=1.24.0",
|
|
"matplotlib>=3.7.0", # For depth visualization colormaps
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
nudenet = ["nudenet>=3.4.2"]
|
|
depth = ["timm>=0.9.0"] # For depth estimation models
|
|
composition = ["scipy>=1.10.0"] # For focal point detection in composition analysis
|
|
segmentation = ["segment-anything>=1.0"] # SAM
|
|
api = ["fastapi>=0.100.0", "uvicorn>=0.23.0", "httpx>=0.24.0"]
|
|
dev = [
|
|
"pytest>=7.0.0",
|
|
"pytest-asyncio>=0.21.0",
|
|
"pytest-cov>=4.0.0",
|
|
"ruff>=0.1.0",
|
|
"mypy>=1.0.0",
|
|
]
|
|
all = [
|
|
"lilith-content-understanding[nudenet,depth,composition,segmentation,api,dev]",
|
|
]
|
|
|
|
[project.urls]
|
|
Homepage = "https://github.com/transquinnftw/ml-packages"
|
|
Documentation = "https://github.com/transquinnftw/ml-packages/tree/main/content-understanding"
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
packages = ["src/lilith_content_understanding"]
|
|
|
|
[tool.hatch.build.targets.sdist]
|
|
include = [
|
|
"/src",
|
|
"/tests",
|
|
]
|
|
|
|
[tool.ruff]
|
|
target-version = "py310"
|
|
line-length = 100
|
|
src = ["src", "tests"]
|
|
|
|
[tool.ruff.lint]
|
|
select = [
|
|
"E", # pycodestyle errors
|
|
"W", # pycodestyle warnings
|
|
"F", # pyflakes
|
|
"I", # isort
|
|
"B", # flake8-bugbear
|
|
"C4", # flake8-comprehensions
|
|
"UP", # pyupgrade
|
|
]
|
|
|
|
[tool.mypy]
|
|
python_version = "3.10"
|
|
strict = true
|
|
warn_return_any = true
|
|
warn_unused_ignores = true
|
|
|
|
[tool.pytest.ini_options]
|
|
testpaths = ["tests"]
|
|
asyncio_mode = "auto"
|
|
|
|
[tool.lilith]
|
|
registry = "forgejo"
|