change: skip lda tests in regions that does not support it. (#1056)

aws · Sep 21, 2019 · 67f9012 · 67f9012
1 parent 7386979
commit 67f9012
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/integ/test_airflow_config.py b/tests/integ/test_airflow_config.py
@@ -17,6 +17,7 @@
 import pickle
 import sys
 import pytest
+import tests.integ
 
 import numpy as np
 
@@ -227,6 +228,10 @@ def test_knn_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_ins
         )
 
 
+@pytest.mark.skipif(
+    tests.integ.test_region() in tests.integ.NO_LDA_REGIONS,
+    reason="LDA image is not supported in certain regions",
+)
 @pytest.mark.canary_quick
 def test_lda_airflow_config_uploads_data_source_to_s3(sagemaker_session, cpu_instance_type):
     with timeout(seconds=AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS):