diff --git a/pkgs/development/python-modules/tritonclient/default.nix b/pkgs/development/python-modules/tritonclient/default.nix new file mode 100644 index 000000000000..f218d02a4e2b --- /dev/null +++ b/pkgs/development/python-modules/tritonclient/default.nix @@ -0,0 +1,65 @@ +{ 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; + }; +} + diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index c65437a7c68d..d0938585db41 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -12617,6 +12617,8 @@ self: super: with self; { trio-websocket = callPackage ../development/python-modules/trio-websocket { }; + tritonclient = callPackage ../development/python-modules/tritonclient { }; + troposphere = callPackage ../development/python-modules/troposphere { }; trove-classifiers = callPackage ../development/python-modules/trove-classifiers { };