minijev-general / code /pyproject.toml
markuz89's picture
MiniJEV v2 model export
45507bb verified
Raw History Blame Contribute Delete
1 kB
[build-system]
requires = ["setuptools>=68"]
build-backend = "setuptools.build_meta"
[project]
name = "minijev"
version = "0.1.0"
description = "MiniJEV: a tiny, calibrated, general-purpose decision model (state + question + choices -> scored choice)"
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
"numpy>=1.26",
"pyyaml>=6",
"psutil>=5.9",
]
[project.optional-dependencies]
# Training / export. Inference-only deployments need just `runtime`.
train = ["torch>=2.3", "onnx>=1.16", "onnxscript>=0.1", "onnxruntime>=1.18", "tensorboard>=2.15"]
runtime = ["onnxruntime>=1.18"]
# Pretrained text encoders (v2 general decision model).
text = ["transformers>=4.40", "tokenizers>=0.19", "huggingface_hub>=0.23"]
dev = ["pytest>=8"]
[project.scripts]
minijev = "minijev.cli:main"
[tool.setuptools.packages.find]
include = ["minijev*", "environments*"]
[tool.setuptools.package-data]
"environments.snake.demo" = ["static/*"]
[tool.pytest.ini_options]
testpaths = ["tests"]