mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-06-10 07:13:38 +00:00
112 lines
1.8 KiB
Nix
112 lines
1.8 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchPypi,
|
|
|
|
# build-system
|
|
hatchling,
|
|
|
|
# dependencies
|
|
braceexpand,
|
|
cloudpickle,
|
|
humanize,
|
|
msgspec,
|
|
overrides,
|
|
packaging,
|
|
pydantic,
|
|
python-dateutil,
|
|
pyyaml,
|
|
requests,
|
|
tenacity,
|
|
urllib3,
|
|
xxhash,
|
|
|
|
# optional-dependencies
|
|
# botocore:
|
|
wrapt,
|
|
# etl:
|
|
aiofiles,
|
|
fastapi,
|
|
flask,
|
|
gunicorn,
|
|
httpx,
|
|
uvicorn,
|
|
# mcp:
|
|
mcp,
|
|
# pytorch:
|
|
alive-progress,
|
|
torch,
|
|
torchdata,
|
|
webdataset,
|
|
}:
|
|
|
|
buildPythonPackage (finalAttrs: {
|
|
pname = "aistore";
|
|
version = "1.25.0";
|
|
pyproject = true;
|
|
__structuredAttrs = true;
|
|
|
|
# Tags on GitHub do not match
|
|
src = fetchPypi {
|
|
inherit (finalAttrs) pname version;
|
|
hash = "sha256-aslNCRSV7QKgvvDuUQPgcbUyUDdGP2kC4ryFu6IVYYE=";
|
|
};
|
|
|
|
build-system = [
|
|
hatchling
|
|
];
|
|
|
|
dependencies = [
|
|
braceexpand
|
|
cloudpickle
|
|
humanize
|
|
msgspec
|
|
overrides
|
|
packaging
|
|
pydantic
|
|
python-dateutil
|
|
pyyaml
|
|
requests
|
|
tenacity
|
|
urllib3
|
|
xxhash
|
|
];
|
|
|
|
optional-dependencies = {
|
|
botocore = [
|
|
wrapt
|
|
];
|
|
etl = [
|
|
aiofiles
|
|
fastapi
|
|
flask
|
|
gunicorn
|
|
httpx
|
|
uvicorn
|
|
];
|
|
mcp = [
|
|
mcp
|
|
];
|
|
pytorch = [
|
|
alive-progress
|
|
torch
|
|
torchdata
|
|
webdataset
|
|
];
|
|
};
|
|
|
|
pythonImportsCheck = [ "aistore" ];
|
|
|
|
# No tests in the Pypi archive
|
|
doCheck = false;
|
|
|
|
meta = {
|
|
description = "Client-side APIs to access and utilize clusters, buckets, and objects on AIStore";
|
|
homepage = "https://aistore.nvidia.com";
|
|
downloadPage = "https://github.com/NVIDIA/aistore/tree/main/python/aistore/sdk";
|
|
changelog = "https://github.com/NVIDIA/aistore/blob/main/python/CHANGELOG.md";
|
|
license = lib.licenses.mit;
|
|
maintainers = with lib.maintainers; [ GaetanLepage ];
|
|
};
|
|
})
|