[project]
name = "er-tp-dgp"
version = "0.1.0"
description = "Event-Reified Temporal Provenance Dual-Granularity Prompting for LLM-based APT detection"
requires-python = ">=3.10"
readme = "README.md"
license = { text = "Research prototype" }
authors = [{ name = "ER-TP-DGP collaborators" }]
dependencies = ["PyYAML>=6.0"]

[project.optional-dependencies]
dev = ["pytest>=7.0"]
local = [
    "torch>=2.3",
    "transformers>=4.45",
    "peft>=0.12",
    "accelerate>=0.34",
    "bitsandbytes>=0.43",
    "datasets>=2.20",
    "numpy>=1.26",
]
embed = [
    "sentence-transformers>=3.0",
    "numpy>=1.26",
]
eval = [
    "scikit-learn>=1.4",
]

[tool.pytest.ini_options]
pythonpath = [".", "src"]
testpaths = ["tests"]

[tool.ruff]
line-length = 100