From 8cdfd08d1c944adcbc0258ea972b097f8c29361f Mon Sep 17 00:00:00 2001 From: Flavia Beo Date: Tue, 28 May 2024 17:33:53 -0300 Subject: [PATCH] Changes metadata field to accept all types Signed-off-by: Flavia Beo --- caikit/interfaces/runtime/data_model/info.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/caikit/interfaces/runtime/data_model/info.py b/caikit/interfaces/runtime/data_model/info.py index 82c5632e8..19d9b7981 100644 --- a/caikit/interfaces/runtime/data_model/info.py +++ b/caikit/interfaces/runtime/data_model/info.py @@ -25,6 +25,7 @@ # Local from caikit.core.data_model import PACKAGE_COMMON, DataObjectBase, dataobject +from caikit.core.data_model.json_dict import JsonDict log = alog.use_channel("RUNTIMEOPS") @@ -61,7 +62,7 @@ class ModelInfo(DataObjectBase): model_path: Annotated[str, FieldNumber(1)] name: Annotated[str, FieldNumber(2)] size: Annotated[int, FieldNumber(3)] - metadata: Annotated[Dict[str, str], FieldNumber(4)] + metadata: Annotated[JsonDict, FieldNumber(4)] loaded: Annotated[bool, FieldNumber(7)] # Module Information