Skip to content

Commit 8249c5a

Browse files
committed
Improve signature def error message
(cherry picked from commit bba5bfc)
1 parent 2ec861b commit 8249c5a

File tree

1 file changed

+1
-1
lines changed
  • pkg/workloads/cortex/tf_api

1 file changed

+1
-1
lines changed

pkg/workloads/cortex/tf_api/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -416,7 +416,7 @@ def exceptions(e):
416416
def extract_signature():
417417
signature_def = local_cache["metadata"]["signatureDef"]
418418
if signature_def.get("predict") is None or signature_def["predict"].get("inputs") is None:
419-
raise UserException("unable to find signature definition for model")
419+
raise UserException('unable to find "predict" in model\'s signature definition')
420420

421421
metadata = {}
422422
for input_name, input_metadata in signature_def["predict"]["inputs"].items():

0 commit comments

Comments
 (0)