[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "openpeer-ntk-trainer"
version = "0.1.0"
description = "OpenPeerLLM trainer using ntkmirror controllers and a tinygrad gate demo"
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
  "torch>=2.2",
  "transformers>=4.42",
  "pandas>=2.2",
  "plotly>=6.0",
  "psutil>=6.0",
]

[project.optional-dependencies]
demo = ["tinygrad>=0.10.0"]
charts = ["openbb>=4.0"]
gui = ["streamlit>=1.36"]
ntk = ["ntkmirror @ git+https://github.com/leochlon/ntkmirror.git"]
all = ["tinygrad>=0.10.0", "openbb>=4.0", "streamlit>=1.36", "ntkmirror @ git+https://github.com/leochlon/ntkmirror.git"]

[project.scripts]
openpeer-trainer = "openpeer_trainer.cli:main"

[tool.setuptools]
package-dir = {"" = "src"}

[tool.setuptools.packages.find]
where = ["src"]