Skip to content

Commit ed290c9

Browse files
sdks/python: address gemini comments
1 parent 8ac2c42 commit ed290c9

File tree

1 file changed

+2
-9
lines changed

1 file changed

+2
-9
lines changed

sdks/python/apache_beam/ml/rag/ingestion/milvus_search_it_test.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -370,10 +370,7 @@ def test_write_on_existent_collection_with_default_schema(self):
370370
for chunk in test_chunks:
371371
self.assertIn(chunk.id, result_by_id)
372372
result_item = result_by_id[chunk.id]
373-
self.assertEqual(
374-
result_item["content"],
375-
chunk.content.text
376-
if hasattr(chunk.content, 'text') else chunk.content)
373+
self.assertEqual(result_item["content"], chunk.content.text)
377374
self.assertEqual(result_item["metadata"], chunk.metadata)
378375

379376
# Verify embedding is present and has correct length.
@@ -388,11 +385,7 @@ def test_write_with_custom_column_specifications(self):
388385

389386
custom_column_specs = [
390387
ColumnSpec("id", int, lambda chunk: int(chunk.id) if chunk.id else 0),
391-
ColumnSpec(
392-
"content",
393-
str, lambda chunk: (
394-
chunk.content.text
395-
if hasattr(chunk.content, 'text') else chunk.content)),
388+
ColumnSpec("content", str, lambda chunk: chunk.content.text),
396389
ColumnSpec("metadata", dict, lambda chunk: chunk.metadata or {}),
397390
ColumnSpec(
398391
"embedding",

0 commit comments

Comments
 (0)