depot/third_party/nixpkgs/pkgs/development/python-modules/tritonclient/default.nix
Default email 9c6ee729d6 Project import generated by Copybara.
GitOrigin-RevId: 6cee3b5893090b0f5f0a06b4cf42ca4e60e5d222
2023-07-15 19:15:38 +02:00

65 lines
1.1 KiB
Nix

{ lib
, fetchPypi
, buildPythonPackage
, pythonOlder
, python-rapidjson
, numpy
# optional dependencies
, grpcio
, packaging
, aiohttp
, geventhttpclient
}:
let
pname = "tritonclient";
version = "2.35.0";
in
buildPythonPackage {
inherit pname version;
format = "wheel";
disabled = pythonOlder "3.7";
src = fetchPypi {
inherit pname version;
python = "py3";
dist = "py3";
format = "wheel";
platform = "manylinux1_x86_64";
hash = "sha256-JXVr+DWkM2g7CiU9STMcZyQJieXgNbKXlX/jqf/oam8=";
};
propagatedBuildInputs = [
python-rapidjson
numpy
];
doCheck = false;
pythonImportsCheck = [
"tritonclient"
];
passthru = {
optional-dependencies = {
http = [
aiohttp
geventhttpclient
];
grpc = [
grpcio
packaging
];
};
};
meta = with lib; {
description = "Triton python client";
homepage = "https://github.com/triton-inference-server/client";
license = licenses.bsd3;
maintainers = with maintainers; [ happysalada ];
platforms = platforms.linux;
};
}