add missing migration for estimated_num_rows (#2950)

* add missing migration for estimated_num_rows * style
huggingface · Jun 25, 2024 · 47b1e5c · 47b1e5c
1 parent 2245d24
commit 47b1e5c
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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",
+            ),
         ]