Skip to content

Commit 6568c04

Browse files
author
David Eigen
committed
fix model param updates
1 parent 2eecf55 commit 6568c04

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

clarifai/client/model.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1218,7 +1218,7 @@ def stream_by_video_file(self,
12181218

12191219
def _get_model_info_for_inference(self, inference_params: Dict = {},
12201220
output_config: Dict = {}) -> None:
1221-
"""Overrides the model version.
1221+
"""Gets the model_info with modified inference params and output config.
12221222
12231223
Args:
12241224
inference_params (dict): The inference params to override.
@@ -1228,13 +1228,11 @@ def _get_model_info_for_inference(self, inference_params: Dict = {},
12281228
select_concepts (list[Concept]): The concepts to select.
12291229
sample_ms (int): The number of milliseconds to sample.
12301230
"""
1231-
if not inference_params and not output_config:
1232-
return self.model_info
1233-
12341231
model_info = resources_pb2.Model()
12351232
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)
1233+
model_info.model_version.output_info.params = inference_params
1234+
model_info.model_version.output_info.output_config.CopyFrom(
1235+
resources_pb2.OutputConfig(**output_config))
12381236
return model_info
12391237

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

0 commit comments

Comments
 (0)