aws
diff --git a/‎test/integration_tests/conftest.py‎
Lines changed: 1 addition & 1 deletion b/‎test/integration_tests/conftest.py‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎…n_tests/cli/test_cli_custom_inference.py‎ ‎…ference/cli/test_cli_custom_inference.py‎test/integration_tests/cli/test_cli_custom_inference.py renamed to test/integration_tests/inference/cli/test_cli_custom_inference.py b/‎…n_tests/cli/test_cli_custom_inference.py‎ ‎…ference/cli/test_cli_custom_inference.py‎test/integration_tests/cli/test_cli_custom_inference.py renamed to test/integration_tests/inference/cli/test_cli_custom_inference.py
diff --git a/‎…ests/cli/test_cli_jumpstart_inference.py‎ ‎…ence/cli/test_cli_jumpstart_inference.py‎test/integration_tests/cli/test_cli_jumpstart_inference.py renamed to test/integration_tests/inference/cli/test_cli_jumpstart_inference.py b/‎…ests/cli/test_cli_jumpstart_inference.py‎ ‎…ence/cli/test_cli_jumpstart_inference.py‎test/integration_tests/cli/test_cli_jumpstart_inference.py renamed to test/integration_tests/inference/cli/test_cli_jumpstart_inference.py
diff --git a/‎…on_tests/sdk/test_sdk_custom_inferece.py‎ ‎…nference/sdk/test_sdk_custom_inferece.py‎test/integration_tests/sdk/test_sdk_custom_inferece.py renamed to test/integration_tests/inference/sdk/test_sdk_custom_inferece.py b/‎…on_tests/sdk/test_sdk_custom_inferece.py‎ ‎…nference/sdk/test_sdk_custom_inferece.py‎test/integration_tests/sdk/test_sdk_custom_inferece.py renamed to test/integration_tests/inference/sdk/test_sdk_custom_inferece.py
diff --git a/‎…ests/sdk/test_sdk_jumpstart_inference.py‎ ‎…ence/sdk/test_sdk_jumpstart_inference.py‎test/integration_tests/sdk/test_sdk_jumpstart_inference.py renamed to test/integration_tests/inference/sdk/test_sdk_jumpstart_inference.py b/‎…ests/sdk/test_sdk_jumpstart_inference.py‎ ‎…ence/sdk/test_sdk_jumpstart_inference.py‎test/integration_tests/sdk/test_sdk_jumpstart_inference.py renamed to test/integration_tests/inference/sdk/test_sdk_jumpstart_inference.py
diff --git a/‎…tests/cli/test_cli_jumpstart_training.py‎ ‎…_tests/training/cli/test_cli_training.py‎test/integration_tests/cli/test_cli_jumpstart_training.py renamed to test/integration_tests/training/cli/test_cli_training.py b/‎…tests/cli/test_cli_jumpstart_training.py‎ ‎…_tests/training/cli/test_cli_training.py‎test/integration_tests/cli/test_cli_jumpstart_training.py renamed to test/integration_tests/training/cli/test_cli_training.py
diff --git a/‎…tests/sdk/test_sdk_jumpstart_training.py‎ ‎…_tests/training/sdk/test_sdk_training.py‎test/integration_tests/sdk/test_sdk_jumpstart_training.py renamed to test/integration_tests/training/sdk/test_sdk_training.py b/‎…tests/sdk/test_sdk_jumpstart_training.py‎ ‎…_tests/training/sdk/test_sdk_training.py‎test/integration_tests/sdk/test_sdk_jumpstart_training.py renamed to test/integration_tests/training/sdk/test_sdk_training.py
@@ -21,7 +21,7 @@ def test_job_name():
 @pytest.fixture(scope="class")
 def image_uri():
     """Return a standard PyTorch image URI for testing."""
-    return "763104351884.dkr.ecr.us-west-2.amazonaws.com/pytorch-training:2.2.0-cpu-py310-ubuntu20.04-sagemaker"
+    return "448049793756.dkr.ecr.us-west-2.amazonaws.com/ptjob:mnist"
 
 @pytest.fixture(scope="class")
 def cluster_name():