diff options
| author | YurenHao0426 <blackhao0426@gmail.com> | 2026-02-15 18:19:50 +0000 |
|---|---|---|
| committer | YurenHao0426 <blackhao0426@gmail.com> | 2026-02-15 18:19:50 +0000 |
| commit | c90b48e3f8da9dd0f8d2ae82ddf977436bb0cfc3 (patch) | |
| tree | 43edac8013fec4e65a0b9cddec5314489b4aafc2 /pyproject.toml | |
Core Hopfield retrieval module with energy-based convergence guarantees,
memory bank, FAISS baseline retriever, evaluation metrics, and end-to-end
pipeline. All 45 tests passing on CPU with synthetic data.
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 36 |
1 files changed, 36 insertions, 0 deletions
diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 0000000..24f7806 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,36 @@ +[build-system] +requires = ["setuptools>=68.0", "wheel"] +build-backend = "setuptools.build_meta" + +[project] +name = "hag" +version = "0.1.0" +requires-python = ">=3.10" +description = "Hopfield-Augmented Generation: RAG with iterative Modern Hopfield Network retrieval" + +dependencies = [ + "torch>=2.0", + "numpy>=1.24", + "faiss-cpu>=1.7", + "transformers>=4.36", + "datasets>=2.14", + "pyyaml>=6.0", + "tqdm>=4.65", +] + +[project.optional-dependencies] +dev = [ + "pytest>=7.0", + "pytest-cov>=4.0", +] +eval = [ + "scikit-learn>=1.3", + "umap-learn>=0.5", + "matplotlib>=3.7", +] + +[tool.setuptools.packages.find] +include = ["hag*"] + +[tool.pytest.ini_options] +testpaths = ["tests"] |
