File tree Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Expand file tree Collapse file tree 1 file changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -981,6 +981,7 @@ def score_model_with_cas(
981981 library_name : str = "Public" ,
982982 model_version : str = "latest" ,
983983 use_cas_gateway : bool = False ,
984+ timeout : int = 300 ,
984985):
985986 score_definition = sd .create_score_definition (
986987 score_def_name ,
@@ -994,7 +995,7 @@ def score_model_with_cas(
994995 use_cas_gateway = use_cas_gateway ,
995996 )
996997 score_execution = se .create_score_execution (score_definition .id )
997- score_execution_poll = se .poll_score_execution_state (score_execution )
998+ score_execution_poll = se .poll_score_execution_state (score_execution , timeout )
998999 print (score_execution_poll )
9991000 score_results = se .get_score_execution_results (score_execution , use_cas_gateway )
10001001 return score_results
You can’t perform that action at this time.
0 commit comments