Skip to content
8 changes: 3 additions & 5 deletions redisvl/migration/async_executor.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,8 @@
from __future__ import annotations

import asyncio
from pathlib import Path

from redisvl.utils.log import get_logger
import time
from pathlib import Path
from typing import Any, AsyncGenerator, Callable, Dict, List, Optional

from redis.exceptions import ResponseError
Expand Down Expand Up @@ -35,6 +33,7 @@
)
from redisvl.redis.utils import array_to_buffer, buffer_to_array
from redisvl.types import AsyncRedisClient
from redisvl.utils.log import get_logger

logger = get_logger(__name__)

Expand Down Expand Up @@ -744,8 +743,7 @@ def _notify(step: str, detail: Optional[str] = None) -> None:
effective_changes = datatype_changes
if has_field_renames:
field_rename_map = {
fr.old_name: fr.new_name
for fr in rename_ops.rename_fields
fr.old_name: fr.new_name for fr in rename_ops.rename_fields
}
effective_changes = {
field_rename_map.get(k, k): v
Expand Down
4 changes: 1 addition & 3 deletions redisvl/migration/batch_planner.py
Original file line number Diff line number Diff line change
Expand Up @@ -233,9 +233,7 @@ def _check_index_applicability(

# Check that add_fields don't already exist.
# Fields being renamed away free their name for new additions.
rename_sources = {
fr.old_name for fr in shared_patch.changes.rename_fields
}
rename_sources = {fr.old_name for fr in shared_patch.changes.rename_fields}
post_rename_fields = (field_names - rename_sources) | rename_target_names
existing_adds: list[str] = []
for field in shared_patch.changes.add_fields:
Expand Down
Loading
Loading