Skip to content

Commit 5f1f873

Browse files
committed
allow for user to access timeout variable in score_model_with_cas
1 parent 57e3214 commit 5f1f873

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/sasctl/tasks.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)