def METHOD_NAME() -> Dict[StorageSetKey, MigrationGroup]:
    result = {}
    for migration_group, _migration_group in _REGISTERED_MIGRATION_GROUPS.items():
        for storage_set_key in _migration_group.storage_set_keys:
            if storage_set_key in result:
                raise DuplicateStorageSetFoundInGroup(
                    f"The storage_set={storage_set_key} is present in more than one MigrationGroup. Each storage_set should only reference one migration group."
                )
            result[storage_set_key] = migration_group
    return result