ClearML/clearml-serving
1
0
Fork 0
mirror of https://github.com/clearml/clearml-serving synced 2025-02-07 13:22:16 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[DEV] feature/bytes-payload | Handle keys when req is bytes

Browse Source
This commit is contained in:
Aleksandar Ivanovski 2022-10-06 15:13:36 +02:00
parent 09ed480bc2
commit 2aa91a3d43
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
clearml_serving/serving/model_request_processor.py
View File

@ -9,6 +9,7 @@ from typing import Optional, Union, Dict, List
import itertools import itertools
import threading import threading
from multiprocessing import Lock from multiprocessing import Lock
from numpy import isin
from numpy.random import choice from numpy.random import choice
from clearml import Task, Model from clearml import Task, Model
@ -1167,7 +1168,7 @@ class ModelRequestProcessor(object):
if metric_endpoint: if metric_endpoint:
metric_keys = set(metric_endpoint.metrics.keys()) metric_keys = set(metric_endpoint.metrics.keys())
# collect inputs # collect inputs
if body: if body and isinstance(body, dict):
keys = set(body.keys()) & metric_keys keys = set(body.keys()) & metric_keys
stats.update({k: body[k] for k in keys}) stats.update({k: body[k] for k in keys})
# collect outputs # collect outputs