diff --git a/clearml_serving/serving/preprocess_service.py b/clearml_serving/serving/preprocess_service.py index a0126d2..2178ffb 100644 --- a/clearml_serving/serving/preprocess_service.py +++ b/clearml_serving/serving/preprocess_service.py @@ -252,7 +252,7 @@ class TritonPreprocessRequest(BasePreprocessRequest): np.uint64: 'uint64_contents', np.int32: 'int_contents', np.uint: 'uint_contents', - np.bool: 'bool_contents', + bool: 'bool_contents', np.float32: 'fp32_contents', np.float64: 'fp64_contents', }