exo-bench was gated behind isDarwin in python/parts.nix because it used exoVenv, which pulls in MLX (Darwin-only). However, exo_bench.py is an HTTP client that only needs loguru, transformers, huggingface-hub, and tiktoken. Made bench a uv workspace member with its own pyproject.toml declaring only the minimal dependencies. Added a separate benchVenv in parts.nix built from that workspace member, and moved exo-bench out of the isDarwin block so it is available on all platforms. Test plan: - `nix run .#exo-bench -- --help` prints argparse help --------- Co-authored-by: Ryuichi Leo Takashige <[email protected]> Co-authored-by: rltakashige <[email protected]>
17 lines
356 B
TOML
17 lines
356 B
TOML
[project]
|
|
name = "exo-bench"
|
|
version = "0.1.0"
|
|
description = "Benchmarking tool for exo distributed inference"
|
|
requires-python = ">=3.13"
|
|
dependencies = [
|
|
"loguru>=0.7.3",
|
|
"transformers>=5.0.0",
|
|
"huggingface-hub>=0.33.4",
|
|
"tiktoken>=0.12.0",
|
|
"jinja2>=3.1.0",
|
|
]
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|