Skip to content

Commit 00e279c

Browse files
Improve pytest latency (#383)
1 parent b0d273e commit 00e279c

File tree

1 file changed

+0
-53
lines changed

1 file changed

+0
-53
lines changed

apis/python/test/test_object_index.py

Lines changed: 0 additions & 53 deletions
Original file line numberDiff line numberDiff line change
@@ -350,29 +350,6 @@ def test_object_index_ivf_flat_cloud(tmp_path):
350350
vector_dim_offset=0,
351351
config=config,
352352
)
353-
# Check that updating the same data doesn't create duplicates
354-
index = object_index.ObjectIndex(uri=index_uri, config=config)
355-
index.update_index(
356-
embeddings_generation_driver_mode=Mode.BATCH,
357-
embeddings_generation_mode=Mode.BATCH,
358-
vector_indexing_mode=Mode.BATCH,
359-
workers=2,
360-
worker_resources=worker_resources,
361-
driver_resources=worker_resources,
362-
kmeans_resources=worker_resources,
363-
ingest_resources=worker_resources,
364-
consolidate_partition_resources=worker_resources,
365-
objects_per_partition=500,
366-
partitions=10,
367-
config=config,
368-
)
369-
evaluate_query(
370-
index_uri=index_uri,
371-
query_kwargs={"nprobe": 10},
372-
dim_id=42,
373-
vector_dim_offset=0,
374-
config=config,
375-
)
376353

377354
# Add new data with a new reader
378355
reader = TestReader(
@@ -403,36 +380,6 @@ def test_object_index_ivf_flat_cloud(tmp_path):
403380
vector_dim_offset=0,
404381
config=config,
405382
)
406-
407-
# Check overwritting existing data
408-
reader = TestReader(
409-
object_id_start=1000,
410-
object_id_end=2000,
411-
vector_dim_offset=1000,
412-
)
413-
index = object_index.ObjectIndex(uri=index_uri, config=config)
414-
index.update_object_reader(reader, config=config)
415-
index.update_index(
416-
embeddings_generation_driver_mode=Mode.BATCH,
417-
embeddings_generation_mode=Mode.BATCH,
418-
vector_indexing_mode=Mode.BATCH,
419-
workers=2,
420-
worker_resources=worker_resources,
421-
driver_resources=worker_resources,
422-
kmeans_resources=worker_resources,
423-
ingest_resources=worker_resources,
424-
consolidate_partition_resources=worker_resources,
425-
objects_per_partition=500,
426-
partitions=10,
427-
config=config,
428-
)
429-
evaluate_query(
430-
index_uri=index_uri,
431-
query_kwargs={"nprobe": 10},
432-
dim_id=2042,
433-
vector_dim_offset=1000,
434-
config=config,
435-
)
436383
delete_uri(index_uri, config)
437384

438385

0 commit comments

Comments
 (0)