Skip to content

tests: unskip three tests in train#5894

Open
lucasjia-aws wants to merge 3 commits into
aws:masterfrom
lucasjia-aws:fix/TestBenchmarkEvaluatorIntegration
Open

tests: unskip three tests in train#5894
lucasjia-aws wants to merge 3 commits into
aws:masterfrom
lucasjia-aws:fix/TestBenchmarkEvaluatorIntegration

Conversation

@lucasjia-aws
Copy link
Copy Markdown
Collaborator

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.

@lucasjia-aws lucasjia-aws changed the title tests: unskip three tests tests: unskip three tests in train May 26, 2026
The MPI driver script used absolute imports (from sagemaker.train.container_drivers...)
which fail at runtime in the training container because sagemaker-train is not installed
there. The driver scripts are copied to /opt/ml/input/data/sm_drivers/ and executed
directly by the container entrypoint.

Changed to sys.path-based relative imports matching the pattern used by
torchrun_driver.py, which works correctly in the container environment.
Remove cross-account dependency in test_benchmark_evaluation_nova_model by
replacing resources from account 052150106756 with our test account
(729646638167) in us-east-1. Also removed mlflow_tracking_server_arn since
no MLflow server exists in us-east-1.

Test remains skipped pending us-east-1 test infrastructure migration to a
dedicated test account.
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