Skip to content

Commit 2aa5e60

Browse files
committed
test: mark bedrock integ tests as serial to avoid concurrent import job quota issues
X-AI-Prompt: add serial marker to bedrock integ tests X-AI-Tool: kiro-cli
1 parent 8164141 commit 2aa5e60

2 files changed

Lines changed: 3 additions & 0 deletions

File tree

‎sagemaker-serve/tests/integ/test_bedrock_provisioned_throughput.py‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ def _setup_model_files(s3_artifacts_uri, s3_client):
119119
logger.warning("Could not create added_tokens.json: %s", e)
120120

121121

122+
@pytest.mark.serial
122123
class TestBedrockImportJobPolling:
123124
"""Test import job polling for OSS models (Option C: deploy only waits for import)."""
124125

@@ -184,6 +185,7 @@ def test_deploy_oss_model_waits_for_import_completion(
184185
assert imported_model_name in model_names
185186

186187

188+
@pytest.mark.serial
187189
class TestBedrockProvisionedThroughput:
188190
"""Test create_provisioned_throughput as a standalone method.
189191

‎sagemaker-serve/tests/integ/test_model_customization_deployment.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -307,6 +307,7 @@ def test_dpo_trainer_build(self, training_job_name, sagemaker_session):
307307
from sagemaker.serve.bedrock_model_builder import BedrockModelBuilder
308308

309309

310+
@pytest.mark.serial
310311
class TestModelCustomizationDeployment:
311312
"""Test suite for deploying fine-tuned models to Bedrock."""
312313

0 commit comments

Comments
 (0)