From 47b1e5c1f8e73fc1fca736a0de411a6398ba6bc2 Mon Sep 17 00:00:00 2001
From: Quentin Lhoest <42851186+lhoestq@users.noreply.github.com>
Date: Tue, 25 Jun 2024 18:28:43 +0200
Subject: [PATCH] add missing migration for estimated_num_rows (#2950)

* add missing migration for estimated_num_rows

* style
---
 jobs/mongodb_migration/src/mongodb_migration/collector.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/jobs/mongodb_migration/src/mongodb_migration/collector.py b/jobs/mongodb_migration/src/mongodb_migration/collector.py
index 2b177f3d6..64cc2c2ef 100644
--- a/jobs/mongodb_migration/src/mongodb_migration/collector.py
+++ b/jobs/mongodb_migration/src/mongodb_migration/collector.py
@@ -81,6 +81,9 @@
 from mongodb_migration.migrations._20240619124500_cache_add_estimated_dataset_info_field_parquet_and_info import (
     MigrationAddEstimatedDatasetInfoToParquetAndInfoCacheResponse,
 )
+from mongodb_migration.migrations._20240624144000_cache_add_estimated_num_rows_field_in_size import (
+    MigrationAddEstimatedNumRowsToSizeCacheResponse,
+)
 from mongodb_migration.renaming_migrations import (
     CacheRenamingMigration,
     QueueRenamingMigration,
@@ -385,4 +388,8 @@ def get_migrations(self) -> list[Migration]:
                 collection=QUEUE_COLLECTION_DATASET_BLOCKAGES,
                 index_name="DATASET_BLOCKAGE_EXPIRE_AFTER_SECONDS",
             ),
+            MigrationAddEstimatedNumRowsToSizeCacheResponse(
+                version="20240624144000",
+                description="add 'estimated_num_rows' field to config-size and dataset-size cache records",
+            ),
         ]