Use gateway with score_model_with_cas call

djm21 · djm21 · commit b3f4358dcc9a · 2024-10-22T17:36:10.000-07:00
diff --git a/src/sasctl/tasks.py b/src/sasctl/tasks.py
@@ -980,6 +980,7 @@ def score_model_with_cas(
     server_name: str = "cas-shared-default",
     library_name: str = "Public",
     model_version: str = "latest",
+    use_cas_gateway: bool = False
 ):
     score_definition = sd.create_score_definition(
         score_def_name,
@@ -990,6 +991,7 @@ def score_model_with_cas(
         server_name=server_name,
         library_name=library_name,
         model_version=model_version,
+        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)
