Merge pull request caikit#723 from flaviabeo/main

Changes metadata field to accept all types
opendatahub-io · May 28, 2024 · 4abc7ba · 4abc7ba
2 parents 06a780e + 8cdfd08
commit 4abc7ba
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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