pyproject.toml

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

[project]
name = "matr"
version = "0.1.0"
# description = ""
readme = "README.md"
license = { text = "Apache 2.0" }
requires-python = ">=3.9.0"
classifiers = [
    "Development Status :: 3 - Alpha",
    "Programming Language :: Python",
]
dependencies = [
    "aiohttp==3.9.1",
    "aiosignal==1.3.1",
    "async-timeout==4.0.3",
    "attrs==23.2.0",
    "certifi==2022.12.7",
    "charset-normalizer==2.1.1",
    "filelock==3.9.0",
    "frozenlist==1.4.1",
    "fsspec==2023.12.2",
    "huggingface-hub==0.20.1",
    "idna==3.4",
    "Jinja2==3.1.2",
    "lightning==2.1.3",
    "lightning-utilities==0.10.0",
    "MarkupSafe==2.1.3",
    "mpmath==1.3.0",
    "multidict==6.0.4",
    "networkx==3.0",
    "numpy==1.24.1",
    "packaging==23.2",
    "Pillow==9.3.0",
    "pytorch-lightning==2.1.3",
    "PyYAML==6.0.1",
    "requests==2.28.1",
    "ruff==0.1.11",
    "safetensors==0.4.1",
    "sympy==1.12",
    "timm==0.9.12",
    "torch==2.1.2+cu118",
    "torchaudio==2.1.2+cu118",
    "torchmetrics==1.2.1",
    "torchvision==0.16.2+cu118",
    "tqdm==4.66.1",
    "triton==2.1.0",
    "typing_extensions==4.4.0",
    "urllib3==1.26.13",
    "xformers==0.0.23.post1",
    "yarl==1.9.4",
]

[tool.black]
line-length = 120

[tool.ruff.isort]

[tool.pytest.ini_options]
testpaths = ["tests"]

[tool.pyright]
include = ["matr"]
exclude = ["**/__pycache__"]
defineConstant = { DEBUG = true }

reportMissingImports = "warning"
reportMissingTypeStubs = false
reportPrivateImportUsage = false

pythonVersion = "3.9"
pythonPlatform = "Linux"

[tool.ruff]
line-length = 120
select = [
    "E",    # pycodestyle errors.
    "F",    # Pyflakes rules.
    "I001", # isort.
    "PLC",  # Pylint convention warnings.
    "PLE",  # Pylint errors.
    "PLR",  # Pylint refactor recommendations.
    "PLW",  # Pylint warnings.
]
ignore = [
    "E501",    # Line too long.
    "E741",    # Ambiguous variable name.
    "F722",    # Forward annotation false positive from jaxtyping. Should be caught by pyright.
    "F821",    # Forward annotation false positive from jaxtyping. Should be caught by pyright.
    "F841",    # Local variable is assigned to but never used.
    "PLR2004", # Magic value used in comparison.
    "PLR0915", # Too many statements.
    "PLR0913", # Too many arguments.
    "PLC0414", # Import alias does not rename variable. (this is used for exporting names)
    "PLC1901", # Use falsey strings.
    "PLR5501", # Use `elif` instead of `else if`.
    "PLR0911", # Too many return statements.
    "PLR0912", # Too many branches.
    "PLW0603", # Globa statement updates are discouraged.
    "PLW2901", # For loop variable overwritten.
]