Skip ap-northeast-2 for TF tests due to instability (#4886)

sirutBuasai · web-flow · commit 53580abf72af · 2025-06-03T22:04:45.000-07:00
diff --git a/test/sagemaker_tests/__init__.py b/test/sagemaker_tests/__init__.py
@@ -184,7 +184,7 @@ def invoke_sm_helper_function(ecr_image, sagemaker_regions, test_function, *test
         )
         # Skip tests if region is ap-northeast-2
         if region == "ap-northeast-2" and "tensorflow" in ecr_image:
-            continue
+            return
         try:
             test_function(tested_ecr_image, sagemaker_session, *test_function_args)
             return

Original file line number	Diff line number	Diff line change
`@@ -184,7 +184,7 @@ def invoke_sm_helper_function(ecr_image, sagemaker_regions, test_function, *test`
`184`	`184`	`)`
`185`	`185`	`# Skip tests if region is ap-northeast-2`
`186`	`186`	`if region == "ap-northeast-2" and "tensorflow" in ecr_image:`
`187`		`- continue`
	`187`	`+ return`
`188`	`188`	`try:`
`189`	`189`	`test_function(tested_ecr_image, sagemaker_session, *test_function_args)`
`190`	`190`	`return`