From 2aa91a3d438e4c9ba68271e3b11addec0f41254e Mon Sep 17 00:00:00 2001
From: Aleksandar Ivanovski <aleksandarivanovski@codechem.com>
Date: Thu, 6 Oct 2022 15:13:36 +0200
Subject: [PATCH] [DEV] feature/bytes-payload | Handle keys when req is bytes

---
 clearml_serving/serving/model_request_processor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/clearml_serving/serving/model_request_processor.py b/clearml_serving/serving/model_request_processor.py
index 084ee1a..67238fc 100644
--- a/clearml_serving/serving/model_request_processor.py
+++ b/clearml_serving/serving/model_request_processor.py
@@ -9,6 +9,7 @@ from typing import Optional, Union, Dict, List
 import itertools
 import threading
 from multiprocessing import Lock
+from numpy import isin
 from numpy.random import choice
 
 from clearml import Task, Model
@@ -1167,7 +1168,7 @@ class ModelRequestProcessor(object):
             if metric_endpoint:
                 metric_keys = set(metric_endpoint.metrics.keys())
                 # collect inputs
-                if body:
+                if body and isinstance(body, dict):
                     keys = set(body.keys()) & metric_keys
                     stats.update({k: body[k] for k in keys})
                 # collect outputs