fix lint

Signed-off-by: yangxuan <[email protected]>
milvus-io · Apr 2, 2024 · 67a7914 · 67a7914
1 parent 273abe5
commit 67a7914
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 7 deletions.
diff --git a/pymilvus/client/entity_helper.py b/pymilvus/client/entity_helper.py
@@ -310,12 +310,20 @@ def pack_field_value_to_field_data(field_value: Any, field_data: Any, field_info
         field_data.vectors.dim = len(field_value) * 8
         field_data.vectors.binary_vector += bytes(field_value)
     elif field_type == DataType.FLOAT16_VECTOR:
-        v_bytes = bytes(field_value) if not isinstance(field_value, np.ndarray) else field_value.view(np.uint8).tobytes()
+        v_bytes = (
+            bytes(field_value)
+            if not isinstance(field_value, np.ndarray)
+            else field_value.view(np.uint8).tobytes()
+        )
 
         field_data.vectors.dim = len(v_bytes) // 2
         field_data.vectors.float16_vector += v_bytes
     elif field_type == DataType.BFLOAT16_VECTOR:
-        v_bytes = bytes(field_value) if not isinstance(field_value, np.ndarray) else field_value.view(np.uint8).tobytes()
+        v_bytes = (
+            bytes(field_value)
+            if not isinstance(field_value, np.ndarray)
+            else field_value.view(np.uint8).tobytes()
+        )
 
         field_data.vectors.dim = len(v_bytes) // 2
         field_data.vectors.bfloat16_vector += v_bytes

diff --git a/pymilvus/client/utils.py b/pymilvus/client/utils.py
@@ -222,7 +222,6 @@ def traverse_rows_info(fields_info: Any, entities: List):
 
         field_name = field["name"]
         location[field_name] = i
-        field_type = field["type"]
 
         if field.get("is_dynamic", False):
             is_dynamic = True

diff --git a/pymilvus/orm/prepare.py b/pymilvus/orm/prepare.py
@@ -17,13 +17,13 @@
 import pandas as pd
 
 from pymilvus.client import entity_helper
+from pymilvus.client.types import DataType
 from pymilvus.exceptions import (
     DataNotMatchException,
     DataTypeNotSupportException,
     ExceptionsMessage,
     UpsertAutoIDTrueException,
 )
-from pymilvus.client.types import DataType
 
 from .schema import CollectionSchema
 
@@ -75,9 +75,9 @@ def prepare_insert_data(
             if isinstance(f_data, np.ndarray):
                 d = f_data.tolist()
 
-            elif (
-                isinstance(f_data[0], np.ndarray)
-                and field.dtype in (DataType.FLOAT16_VECTOR, DataType.BFLOAT16_VECTOR)
+            elif isinstance(f_data[0], np.ndarray) and field.dtype in (
+                DataType.FLOAT16_VECTOR,
+                DataType.BFLOAT16_VECTOR,
             ):
                 d = [bytes(ndarr.view(np.uint8).tolist()) for ndarr in f_data]