pyproject.toml

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

[project]
name = "ckatorch"
authors = [
  {name = "Alessandro Ristori", email = "aleristori97@gmail.com"}
]
version = "1.0.0"
description = "Centered Kernel Alignment in PyTorch"
readme = "README.md"
license = {file = "LICENSE"}
requires-python = ">=3.10"
keywords = ["CKA", "Centered Kernel Aligment", "Neural Network Similarity", "Deep Learning"]
classifiers = [
  "License :: OSI Approved :: MIT License",
  "Programming Language :: Python :: 3.10",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Programming Language :: Python :: 3 :: Only",
]
dependencies = [
    "matplotlib>=3.8.0",
    "seaborn>=0.12.2",
    "torch>=2.3.1",
    "torchvision>=0.18.1",
    "tqdm>=4.66.3",
    "transformers>=4.43.0",
]


[project.optional-dependencies]
dev = [
    "pre-commit>=3.7.1",
    "ruff>=0.6.5",
    "typos>=1.24.5",
]

[project.urls]
Repository = "https://github.com/RistoAle97/centered-kernel-alignment"

[tool.ruff]
line-length = 120
output-format = "grouped"
target-version = "py310"

[tool.ruff.lint]
ignore = [
    "E111", # Indentation is not a multiple of {indent_width}
    "E114", # Indentation is not a multiple of {indent_width} (comment)
    "E117", # Over-indented (comment)
    "E501", # Line too long
]
select = [
    "ARG", # Flake8 unused arguments
    "D", # Pydocstyle errors
    "E", # PEP8 errors
    "F", # PEP8 formatting
    "I", # Import sorting
    "N", # PEP8 naming
    "RUF", # Ruff specific rules
    "UP", # Pyupgrade upgrades
]

[tool.ruff.lint.extend-per-file-ignores]
"examples/*" = ["D"]

[tool.ruff.lint.isort]
force-single-line = false
lines-after-imports = 2

[tool.ruff.lint.pydocstyle]
convention = "google"

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

[tool.uv.pip]
extra-index-url = ["https://download.pytorch.org/whl/cu124"]