pyproject.toml

[project]
name = "interprot"
version = "0.1.0"
description = "Mechanistic interpretability tools for protein language models"
authors = [
    {name = "Etowah Adams", email = "etowahadams@gmail.com"},
    {name = "Liam Bai", email = "liambai2000@gmail.com"}
]
dependencies = [
    "click",
    "transformers",
    "torch>=2.2.0",
    "polars",
    "numpy<2",
    "scikit-learn",
    "tqdm",
    "pandas",
    "scipy"
]
requires-python = ">=3.10"
readme = "README.md"

[tool.setuptools]
packages = [
    "interprot",
    "interprot.autointerp",
    "interprot.oned_probe",
    "interprot.make_viz_files",
]

[project.scripts]
autointerp = "interprot.autointerp.__main__:cli"
oned_probe = "interprot.oned_probe.__main__:cli"
make_viz_files = "interprot.make_viz_files.__main__:make_viz_files"

[project.urls]
Homepage = "https://github.com/etowahadams/interprot"

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.ruff]
line-length = 100
target-version = "py310"

[tool.ruff.lint]
select = ["E", "F", "I"]
ignore = []

[tool.ruff.format]
quote-style = "double"
indent-style = "space"
skip-magic-trailing-comma = false
line-ending = "auto"

[project.optional-dependencies]
dev = [
    "pytest",
    "hatch",
]

[tool.hatch.build.targets.wheel]
packages = [
    "interprot",
    "interprot.autointerp",
    "interprot.oned_probe"
]