pyproject.toml

[build-system]
requires = ['poetry-core>=1.0.0']
build-backend = 'poetry.core.masonry.api'

[tool.poetry]
name = 'vsgan'
version = '1.6.4'
description = 'PyTorch-based Super-Resolution and Restoration Image Processing Module for VapourSynth'
authors = ['rlaphoenix <rlaphoenix@pm.me>']
license = 'MIT'
readme = 'README.md'
homepage = 'https://github.com/rlaphoenix/vsgan'
repository = 'https://github.com/rlaphoenix/vsgan'
keywords = ['vapoursynth', 'upscaling', 'gan', 'deep-learning', 'esrgan']
classifiers = [
    'Development Status :: 5 - Production/Stable',
    'Intended Audience :: Developers',
    'Natural Language :: English',
    'Operating System :: OS Independent',
    'Topic :: Multimedia :: Video',
]

[tool.poetry.dependencies]
python = ">=3.9,<3.12"
numpy = "^1.26.0"
einops = "^0.6.1"
VapourSynth = [
    # https://github.com/vapoursynth/vapoursynth/issues/858
    {version = "57", optional = true, python = "~3.9"},
    {version = "61", optional = true, python = "~3.10"},
    {version = "62", optional = true, python = "^3.11"},
]
torch = {version = ">=1.12.0", optional = true}
Sphinx = {version = "^7.2.6", optional = true}
furo = {version = "^2023.9.10", optional = true}
dunamai = {version = "^1.18.0", optional = true}
m2r2 = {version = "^0.3.3", optional = true}
sphinxcontrib-youtube = {version = "^1.4.0", optional = true}
sphinxcontrib-images = {version = "^0.9.4", optional = true}

[tool.poetry.extras]
vs = ["VapourSynth"]
pytorch = ["torch"]
docs = ["Sphinx", "furo", "dunamai", "m2r2", "sphinxcontrib-youtube", "sphinxcontrib-images"]

[tool.poetry.dev-dependencies]
pre-commit = "^3.4.0"
isort = "^5.12.0"
flake8 = "^6.1.0"

[tool.isort]
line_length = 120