Skip to content

Commit 8c02e44

Browse files
fix milvus test
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
1 parent 526981e commit 8c02e44

File tree

1 file changed

+9
-2
lines changed
  • sdk/python/feast/infra/online_stores/milvus_online_store

1 file changed

+9
-2
lines changed

sdk/python/feast/infra/online_stores/milvus_online_store/milvus.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -292,8 +292,15 @@ def online_write_batch(
292292
}
293293
# Filter out timestamp fields that are handled separately
294294
filtered_values_dict = {
295-
k: v for k, v in values_dict.items()
296-
if k not in ["event_timestamp", "created_timestamp", "event_ts", "created_ts"]
295+
k: v
296+
for k, v in values_dict.items()
297+
if k
298+
not in [
299+
"event_timestamp",
300+
"created_timestamp",
301+
"event_ts",
302+
"created_ts",
303+
]
297304
}
298305
single_entity_record.update(filtered_values_dict)
299306
# Ensure all required fields exist, setting missing ones to empty strings

0 commit comments

Comments
 (0)