From 083635c8035c777a92b1831ad7cb2c11fa644028 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Sat, 23 Sep 2023 17:36:28 +0300
Subject: [PATCH] Add `str` type to Triton type conversion

---
 clearml_serving/engines/triton/triton_helper.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/clearml_serving/engines/triton/triton_helper.py b/clearml_serving/engines/triton/triton_helper.py
index 533d06a..ff606cb 100644
--- a/clearml_serving/engines/triton/triton_helper.py
+++ b/clearml_serving/engines/triton/triton_helper.py
@@ -422,6 +422,8 @@ class TritonHelper(object):
             return "FP32"
         elif np_dtype == np.float64:
             return "FP64"
+        elif np_dtype == str:
+            return "STRING"
         elif np_dtype == np.object_ or np_dtype.type == np.bytes_:
             return "BYTES"
         return None