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(
981
981
library_name : str = "Public" ,
982
982
model_version : str = "latest" ,
983
983
use_cas_gateway : bool = False ,
984
+ timeout : int = 300 ,
984
985
):
985
986
score_definition = sd .create_score_definition (
986
987
score_def_name ,
@@ -994,7 +995,7 @@ def score_model_with_cas(
994
995
use_cas_gateway = use_cas_gateway ,
995
996
)
996
997
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 )
998
999
print (score_execution_poll )
999
1000
score_results = se .get_score_execution_results (score_execution , use_cas_gateway )
1000
1001
return score_results
You can’t perform that action at this time.
0 commit comments