From 69a9fcd0cee94da94b6ae121b4859b7b7af1ebb6 Mon Sep 17 00:00:00 2001
From: Kalyani Nikure <110067132+knikure@users.noreply.github.com>
Date: Tue, 5 Mar 2024 07:05:42 -0800
Subject: [PATCH] fix: Skip No Canvas regions for test_deploy_best_candidate
 (#4477)

---
 tests/integ/__init__.py        | 7 +++++++
 tests/integ/test_auto_ml_v2.py | 3 ++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/tests/integ/__init__.py b/tests/integ/__init__.py
index fbf32c3acf..434f4dd744 100644
--- a/tests/integ/__init__.py
+++ b/tests/integ/__init__.py
@@ -139,6 +139,13 @@
     "af-south-1",
     "eu-south-1",
 ]
+NO_CANVAS_REGIONS = [
+    "ca-central-1",
+    "eu-north-1",
+    "eu-west-2",
+    "sa-east-1",
+    "us-west-1",
+]
 NO_MODEL_MONITORING_REGIONS = ["me-south-1", "af-south-1", "eu-south-1"]
 DRIFT_CHECK_BASELINES_SUPPORTED_REGIONS = [
     "us-east-2",
diff --git a/tests/integ/test_auto_ml_v2.py b/tests/integ/test_auto_ml_v2.py
index 7749b0c5f2..91802770d1 100644
--- a/tests/integ/test_auto_ml_v2.py
+++ b/tests/integ/test_auto_ml_v2.py
@@ -330,7 +330,8 @@ def test_best_candidate(
 
 
 @pytest.mark.skipif(
-    tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS,
+    tests.integ.test_region() in tests.integ.NO_AUTO_ML_REGIONS
+    or tests.integ.test_region() in tests.integ.NO_CANVAS_REGIONS,
     reason="AutoML is not supported in the region yet.",
 )
 @pytest.mark.release