1{
2 lib,
3 fetchPypi,
4 buildPythonPackage,
5 pythonOlder,
6 python-rapidjson,
7 numpy,
8 # optional dependencies
9 grpcio,
10 packaging,
11 aiohttp,
12 geventhttpclient,
13}:
14
15let
16 pname = "tritonclient";
17 version = "2.35.0";
18in
19buildPythonPackage {
20 inherit pname version;
21 format = "wheel";
22
23 disabled = pythonOlder "3.7";
24
25 src = fetchPypi {
26 inherit pname version;
27 python = "py3";
28 dist = "py3";
29 format = "wheel";
30 platform = "manylinux1_x86_64";
31 hash = "sha256-JXVr+DWkM2g7CiU9STMcZyQJieXgNbKXlX/jqf/oam8=";
32 };
33
34 propagatedBuildInputs = [
35 python-rapidjson
36 numpy
37 ];
38
39 doCheck = false;
40
41 pythonImportsCheck = [ "tritonclient" ];
42
43 passthru = {
44 optional-dependencies = {
45 http = [
46 aiohttp
47 geventhttpclient
48 ];
49 grpc = [
50 grpcio
51 packaging
52 ];
53 };
54 };
55
56 meta = with lib; {
57 description = "Triton python client";
58 homepage = "https://github.com/triton-inference-server/client";
59 license = licenses.bsd3;
60 maintainers = with maintainers; [ happysalada ];
61 platforms = platforms.linux;
62 };
63}