From 5f1f8730203a59d72a6963cdf0940fe6b1dd733a Mon Sep 17 00:00:00 2001 From: djm21 Date: Fri, 20 Jun 2025 08:59:35 -0700 Subject: [PATCH 1/2] allow for user to access timeout variable in score_model_with_cas --- src/sasctl/tasks.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/sasctl/tasks.py b/src/sasctl/tasks.py index 6bf5c67b..ca659630 100644 --- a/src/sasctl/tasks.py +++ b/src/sasctl/tasks.py @@ -981,6 +981,7 @@ def score_model_with_cas( library_name: str = "Public", model_version: str = "latest", use_cas_gateway: bool = False, + timeout: int = 300, ): score_definition = sd.create_score_definition( score_def_name, @@ -994,7 +995,7 @@ def score_model_with_cas( use_cas_gateway=use_cas_gateway, ) score_execution = se.create_score_execution(score_definition.id) - score_execution_poll = se.poll_score_execution_state(score_execution) + score_execution_poll = se.poll_score_execution_state(score_execution, timeout) print(score_execution_poll) score_results = se.get_score_execution_results(score_execution, use_cas_gateway) return score_results From e8129e26d43249b1942d1018312c73a1bc19988d Mon Sep 17 00:00:00 2001 From: djm21 Date: Fri, 27 Jun 2025 10:53:52 -0700 Subject: [PATCH 2/2] update create_score_definition to work with uuids --- src/sasctl/_services/score_definitions.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/sasctl/_services/score_definitions.py b/src/sasctl/_services/score_definitions.py index 448a28c5..2c05611f 100644 --- a/src/sasctl/_services/score_definitions.py +++ b/src/sasctl/_services/score_definitions.py @@ -83,9 +83,7 @@ def create_score_definition( else: object_descriptor_type = "sas.models.model.ds2" - if cls._model_repository.is_uuid(model): - model_id = model - elif isinstance(model, dict) and "id" in model: + if isinstance(model, dict) and "id" in model: model_id = model["id"] else: model = cls._model_repository.get_model(model)