fix readiness state check (#5922)

getsentry · May 16, 2024 · d4859d4 · d4859d4
1 parent 64c044e
commit d4859d4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/snuba/migrations/groups.py b/snuba/migrations/groups.py
@@ -204,7 +204,9 @@ def get_storage_set_keys(group: MigrationGroup) -> Set[StorageSetKey]:
 def get_group_readiness_state_from_storage_set(
     storage_set_key: StorageSetKey,
 ) -> ReadinessState:
-    migration_group = _STORAGE_SET_TO_MIGRATION_GROUP_MAPPING[storage_set_key]
+    migration_group = _STORAGE_SET_TO_MIGRATION_GROUP_MAPPING.get(storage_set_key, None)
+    if not migration_group:
+        return ReadinessState.LIMITED
     registered_migration_group = _REGISTERED_MIGRATION_GROUPS.get(migration_group, None)
     if registered_migration_group:
         return registered_migration_group.readiness_state