refiners/pyproject.toml

173 lines
4.3 KiB
TOML
Raw Normal View History

[project]
2023-08-04 13:28:41 +00:00
name = "refiners"
2024-08-02 18:11:05 +00:00
dynamic = ["version"]
2024-02-01 17:31:50 +00:00
description = "The simplest way to train and run adapters on top of foundation models"
authors = [{ name = "The Finegrain Team", email = "bonjour@lagon.tech" }]
license = { text = "MIT License" }
dependencies = [
"torch>=2.1.1",
"safetensors>=0.4.5",
2024-07-17 16:57:41 +00:00
"pillow>=10.4.0",
"jaxtyping>=0.2.23",
"packaging>=23.2",
2024-04-02 15:59:03 +00:00
"numpy>=1.26.4",
]
2023-08-04 13:28:41 +00:00
readme = "README.md"
requires-python = ">= 3.10"
classifiers = [
"Typing :: Typed",
"Intended Audience :: Developers",
"Intended Audience :: Science/Research",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Software Development :: Libraries :: Python Modules",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"License :: OSI Approved :: MIT License",
]
2023-08-04 13:28:41 +00:00
2024-08-02 18:11:05 +00:00
[project.urls]
source = "https://github.com/finegrain-ai/refiners"
documentation = "https://refine.rs/"
[project.optional-dependencies]
training = [
"bitsandbytes>=0.41.2.post2",
"pydantic>=2.5.2",
"prodigyopt>=1.0",
"torchvision>=0.16.1",
"loguru>=0.7.2",
"wandb>=0.16.0",
2024-05-28 14:24:16 +00:00
"neptune>=1.10.4",
"datasets>=2.15.0",
2023-12-15 17:02:32 +00:00
"tomli>=2.0.1",
"gitpython>=3.1.43",
]
test = [
"diffusers>=0.26.1",
"transformers>=4.35.2",
"piq>=0.8.0",
"torchvision>=0.16.1",
# An unofficial Python package for Meta AI's Segment Anything Model:
# https://github.com/opengeos/segment-anything
"segment-anything-py>=1.0",
2024-03-21 13:59:36 +00:00
# Official Python package for HQ-SAM
"segment-anything-hq>=0.3",
# HQ-SAM missing dependency:
# https://github.com/SysCV/sam-hq/pull/59
"timm>=0.5.0",
"sentencepiece>=0.2.0",
]
conversion = [
"diffusers>=0.26.1",
"transformers>=4.35.2",
"segment-anything-py>=1.0",
"requests>=2.26.0",
"tqdm>=4.62.3",
"gdown>=5.2.0",
]
doc = [
# required by mkdocs to format the signatures
"black>=24.1.1",
"mkdocs-material>=9.5.6",
"mkdocstrings[python]>=0.24.0",
"mkdocs-literate-nav>=0.6.1",
]
2024-08-27 09:44:44 +00:00
solutions = [
"huggingface-hub>=0.24.6",
]
2023-08-04 13:28:41 +00:00
[build-system]
2024-08-02 18:11:05 +00:00
requires = ["hatchling", "hatch-vcs"]
build-backend = "hatchling.build"
2023-08-04 13:28:41 +00:00
2024-08-02 18:11:05 +00:00
[tool.hatch.version]
source = "vcs"
fallback-version = '0.0.0'
2023-08-04 13:28:41 +00:00
[tool.rye]
managed = true
dev-dependencies = [
2024-07-17 16:57:41 +00:00
"pyright==1.1.372",
"docformatter>=1.7.5",
"pytest>=8.0.0",
2024-01-27 15:10:31 +00:00
"coverage>=7.4.1",
"typos>=1.18.2",
2024-09-05 09:30:48 +00:00
"comfy-cli>=1.1.6",
]
[tool.hatch.metadata]
allow-direct-references = true
[tool.rye.scripts]
serve-docs = "mkdocs serve"
2024-01-27 18:32:40 +00:00
test-cov = "coverage run -m pytest"
# Work around for "Couldn't parse" errors due to e.g. opencv-python:
# https://github.com/nedbat/coveragepy/issues/1653
build-html-cov = { cmd = "coverage html", env = { PYTHONWARNINGS = "ignore:Couldn't parse::coverage.report_core" } }
2024-01-31 18:11:34 +00:00
serve-cov-report = { chain = [
"build-html-cov",
"python -m http.server 8080 -b 127.0.0.1 -d htmlcov",
] }
2023-08-04 13:28:41 +00:00
[tool.black]
line-length = 120
[tool.ruff]
src = ["src"] # see https://docs.astral.sh/ruff/settings/#src
2024-02-02 13:17:28 +00:00
line-length = 120
[tool.ruff.lint]
select = [
2024-01-27 18:33:55 +00:00
"I", # isort
]
2023-08-04 13:28:41 +00:00
ignore = [
"F722", # forward-annotation-syntax-error, because of Jaxtyping
"E731", # do-not-assign-lambda
2023-08-04 13:28:41 +00:00
]
[tool.ruff.lint.isort]
# Allow this kind of import on a single line:
#
# from torch import device as Device, dtype as DType
#
combine-as-imports = true
[tool.docformatter]
black = true
2023-08-04 13:28:41 +00:00
[tool.pyright]
include = ["src/refiners", "tests", "scripts"]
2023-08-16 09:17:55 +00:00
strict = ["*"]
exclude = ["**/__pycache__", "tests/weights"]
2023-08-04 13:28:41 +00:00
reportMissingTypeStubs = "warning"
2024-01-27 18:32:40 +00:00
[tool.coverage.run]
branch = true
source = ["src/refiners"]
2024-01-27 18:32:40 +00:00
# Also apply to HTML output, where appropriate
[tool.coverage.report]
ignore_errors = true # see `build-html-cov` for details
exclude_also = [
"def __repr__",
"raise NotImplementedError",
"if TYPE_CHECKING:",
"class .*\\bProtocol\\):",
"@(abc\\.)?abstractmethod",
]
[tool.typos.default]
extend-ignore-identifiers-re = ["NDArray*", "interm", "af000ded"]
2024-03-21 13:59:36 +00:00
[tool.typos.default.extend-words]
adaptee = "adaptee" # Common name for an adapter's target
2024-03-21 13:59:36 +00:00
[tool.pytest.ini_options]
filterwarnings = [
"ignore::UserWarning:segment_anything_hq.modeling.tiny_vit_sam.*",
"ignore::DeprecationWarning:timm.models.layers.*",
"ignore::DeprecationWarning:timm.models.registry.*"
]
addopts = "--import-mode=importlib"