Experiment config updates

mitdbg · Nov 29, 2023 · 652209d · 652209d
1 parent 7ccdc00
commit 652209d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
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
@@ -120,6 +120,8 @@ def main():
     new_placement = {}
     for table in blueprint.tables():
         new_placement[table.name] = [Engine.Aurora, Engine.Athena]
+        if table.name == "telemetry":
+            new_placement[table.name] = [Engine.Athena]
     enum_blueprint.set_table_locations(new_placement)
 
     # 6. Transition to the new blueprint.

diff --git a/src/brad/utils/table_sizer.py b/src/brad/utils/table_sizer.py
@@ -63,7 +63,7 @@ async def table_size_rows(
             cursor = conn.cursor_sync()
             await cursor.execute(query)
             row = cursor.fetchone_sync()
-            assert row is not None
+            assert row is not None, table_name
             result = int(row[0])
             if result < 0:
                 logger.warning(
@@ -80,7 +80,7 @@ async def table_size_rows(
         cursor = conn.cursor_sync()
         await cursor.execute(query)
         row = cursor.fetchone_sync()
-        assert row is not None
+        assert row is not None, table_name
         result = int(row[0])
         return result