Skip to content

test: Mark capacity-flaky GPU integ tests as slow_test#5998

Open
lucasjia-aws wants to merge 1 commit into
aws:master-v2from
lucasjia-aws:fix/v2-tests
Open

test: Mark capacity-flaky GPU integ tests as slow_test#5998
lucasjia-aws wants to merge 1 commit into
aws:master-v2from
lucasjia-aws:fix/v2-tests

Conversation

@lucasjia-aws

Copy link
Copy Markdown
Collaborator

Mark five integ tests that require large GPU instances (g5/g4dn) or LMI/JumpStart LLM endpoints as slow_test so they are excluded from the default CI gate (-m "not ... and not slow_test"). These tests repeatedly fail with InsufficientInstanceCapacity or endpoint-creation timeouts due to regional GPU capacity shortages rather than SDK regressions:

  • test_huggingface_torch_distributed_g5_glue
  • test_model_builder_happy_path_with_task_provided_local_schema_mode
  • test_tei_sagemaker_endpoint
  • test_model_package_arn_jumpstart_model
  • test_jumpstart_hub_gated_model

Also add the missing pytest import in test_huggingface_torch_distributed.py.

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

Mark five integ tests that require large GPU instances (g5/g4dn) or
LMI/JumpStart LLM endpoints as slow_test so they are excluded from the
default CI gate (-m "not ... and not slow_test"). These tests repeatedly
fail with InsufficientInstanceCapacity or endpoint-creation timeouts due
to regional GPU capacity shortages rather than SDK regressions:

- test_huggingface_torch_distributed_g5_glue
- test_model_builder_happy_path_with_task_provided_local_schema_mode
- test_tei_sagemaker_endpoint
- test_model_package_arn_jumpstart_model
- test_jumpstart_hub_gated_model

Also add the missing pytest import in test_huggingface_torch_distributed.py.
@lucasjia-aws lucasjia-aws requested a review from a team as a code owner July 7, 2026 22:14
@lucasjia-aws lucasjia-aws requested a review from chad119 July 7, 2026 22:14
@lucasjia-aws lucasjia-aws deployed to auto-approve July 7, 2026 22:14 — with GitHub Actions Active
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant