We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 57e3214 commit 5f1f873Copy full SHA for 5f1f873
1 file changed
src/sasctl/tasks.py
@@ -981,6 +981,7 @@ def score_model_with_cas(
981
library_name: str = "Public",
982
model_version: str = "latest",
983
use_cas_gateway: bool = False,
984
+ timeout: int = 300,
985
):
986
score_definition = sd.create_score_definition(
987
score_def_name,
@@ -994,7 +995,7 @@ def score_model_with_cas(
994
995
use_cas_gateway=use_cas_gateway,
996
)
997
score_execution = se.create_score_execution(score_definition.id)
- score_execution_poll = se.poll_score_execution_state(score_execution)
998
+ score_execution_poll = se.poll_score_execution_state(score_execution, timeout)
999
print(score_execution_poll)
1000
score_results = se.get_score_execution_results(score_execution, use_cas_gateway)
1001
return score_results
0 commit comments