From 1a60be55c3b0deec96c1bd89c62154b9b7cc4c1d Mon Sep 17 00:00:00 2001
From: happysalada <raphael@megzari.com>
Date: Thu, 6 Jul 2023 08:31:35 +0800
Subject: [PATCH] python310Packages.tritonclient: init at 2.35.0

---
 .../python-modules/tritonclient/default.nix   | 65 +++++++++++++++++++
 pkgs/top-level/python-packages.nix            |  2 +
 2 files changed, 67 insertions(+)
 create mode 100644 pkgs/development/python-modules/tritonclient/default.nix

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 { };