Skip to content

Commit 162d81d

Browse files
author
Kevin Kim
committed
Fix datarow test due to precomputedEmbeddings field automatically added
1 parent 7b2bbb9 commit 162d81d

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

tests/integration/test_data_rows.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -313,8 +313,13 @@ def test_create_data_rows_with_metadata(dataset, image_url):
313313
assert requests.get(image_url).content == \
314314
requests.get(row.row_data).content
315315
assert row.media_attributes is not None
316-
assert len(row.metadata_fields) == 4
317-
assert [m["schemaId"] for m in row.metadata_fields
316+
317+
# Remove 'precomputedImageEmbedding' metadata if automatically added
318+
filtered_md_fields = list(
319+
filter(lambda md: md["name"] != "precomputedImageEmbedding",
320+
row.metadata_fields))
321+
assert len(filtered_md_fields) == 4
322+
assert [m["schemaId"] for m in filtered_md_fields
318323
].sort() == EXPECTED_METADATA_SCHEMA_IDS
319324

320325

0 commit comments

Comments
 (0)