mirror of
https://github.com/clearml/clearml-serving
synced 2025-02-07 05:18:12 +00:00
[DEV] feature/bytes-payload | Handle keys when req is bytes
This commit is contained in:
parent
09ed480bc2
commit
2aa91a3d43
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user