Skip to content

Commit 8b2c166

Browse files
committed
Update score_chain.jinja2 to call invoke instead of run.
1 parent 579bd43 commit 8b2c166

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ads/llm/templates/score_chain.jinja2

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,5 +165,5 @@ def predict(data, model=load_model(), input_schema_path=os.path.join(os.path.dir
165165

166166
"""
167167
features = pre_inference(data, input_schema_path)
168-
yhat = post_inference(model.run(features))
168+
yhat = post_inference(model.invoke(features))
169169
return {'prediction': yhat}

0 commit comments

Comments
 (0)