fix: executor minor cleanups (#561)

- Fix _quantize_vectors docstring: 'documents quantized' not 'processed'
- Wrap test_migration_v1 assertions in try/finally for cleanup

Author: nkanu17

redisvl/migration/executor.py

@@ -893,7 +893,7 @@ def _quantize_vectors(
             checkpoint_path: Optional path for checkpoint file (enables resume)
 
         Returns:
-            Number of documents processed
+            Number of documents quantized
         """
         client = source_index._redis_client
         total_keys = len(keys)

tests/integration/test_migration_v1.py

@@ -102,26 +102,28 @@ def test_drop_recreate_plan_apply_validate_flow(redis_url, worker_id, tmp_path):
         query_check_file=str(query_check_path),
     )
 
-    assert report.result == "succeeded"
-    assert report.validation.schema_match is True
-    assert report.validation.doc_count_match is True
-    assert report.validation.key_sample_exists is True
-    assert report.validation.indexing_failures_delta == 0
-    assert not report.validation.errors
-    assert report.benchmark_summary.documents_indexed_per_second is not None
+    try:
+        assert report.result == "succeeded"
+        assert report.validation.schema_match is True
+        assert report.validation.doc_count_match is True
+        assert report.validation.key_sample_exists is True
+        assert report.validation.indexing_failures_delta == 0
+        assert not report.validation.errors
+        assert report.benchmark_summary.documents_indexed_per_second is not None
 
-    live_index = SearchIndex.from_existing(index_name, redis_url=redis_url)
-    assert schemas_equal(live_index.schema.to_dict(), plan.merged_target_schema)
+        live_index = SearchIndex.from_existing(index_name, redis_url=redis_url)
+        assert schemas_equal(live_index.schema.to_dict(), plan.merged_target_schema)
 
-    validator = MigrationValidator()
-    validation, _target_info, _duration = validator.validate(
-        load_migration_plan(str(plan_path)),
-        redis_url=redis_url,
-        query_check_file=str(query_check_path),
-    )
-    assert validation.schema_match is True
-    assert validation.doc_count_match is True
-    assert validation.key_sample_exists is True
-    assert not validation.errors
-
-    live_index.delete(drop=True)
+        validator = MigrationValidator()
+        validation, _target_info, _duration = validator.validate(
+            load_migration_plan(str(plan_path)),
+            redis_url=redis_url,
+            query_check_file=str(query_check_path),
+        )
+        assert validation.schema_match is True
+        assert validation.doc_count_match is True
+        assert validation.key_sample_exists is True
+        assert not validation.errors
+    finally:
+        live_index = SearchIndex.from_existing(index_name, redis_url=redis_url)
+        live_index.delete(drop=True)