Skip to content

Commit 14427b4

Browse files
author
David Eigen
committed
fix model param updates
1 parent 2eecf55 commit 14427b4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

clarifai/client/model.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1233,8 +1233,9 @@ def _get_model_info_for_inference(self, inference_params: Dict = {},
12331233

12341234
model_info = resources_pb2.Model()
12351235
model_info.CopyFrom(self.model_info)
1236-
model_info.model_version.output_info.params.update(inference_params)
1237-
model_info.model_version.output_info.output_config.update(output_config)
1236+
model_info.model_version.output_info.params = inference_params
1237+
model_info.model_version.output_info.output_config.CopyFrom(
1238+
resources_pb2.OutputConfig(**output_config))
12381239
return model_info
12391240

12401241
def _list_concepts(self) -> List[str]:

0 commit comments

Comments
 (0)