From 657133805e65877e09bf7c76051b199517001a8d Mon Sep 17 00:00:00 2001
From: Geoffrey Yu <geoffxy@mit.edu>
Date: Wed, 29 Nov 2023 16:47:40 +0000
Subject: [PATCH] Fix placement

---
 .../15-e2e-scenarios-v2/specialized/set_up_vector_blueprint.py  | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/experiments/15-e2e-scenarios-v2/specialized/set_up_vector_blueprint.py b/experiments/15-e2e-scenarios-v2/specialized/set_up_vector_blueprint.py
index 5d98884f..d46ff411 100644
--- a/experiments/15-e2e-scenarios-v2/specialized/set_up_vector_blueprint.py
+++ b/experiments/15-e2e-scenarios-v2/specialized/set_up_vector_blueprint.py
@@ -122,6 +122,8 @@ def main():
         new_placement[table.name] = [Engine.Aurora, Engine.Athena]
         if table.name == "telemetry":
             new_placement[table.name] = [Engine.Athena]
+        if table.name == "embeddings":
+            new_placement[table.name] = [Engine.Aurora]
     enum_blueprint.set_table_locations(new_placement)
 
     # 6. Transition to the new blueprint.