From 3308b2be51746e3f21a5183aa7a635f5e44efc1a Mon Sep 17 00:00:00 2001 From: Xuanwo Date: Tue, 27 Jan 2026 19:12:02 +0800 Subject: [PATCH] ci: fix benchmark regress --- python/python/ci_benchmarks/datagen/wikipedia.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/python/ci_benchmarks/datagen/wikipedia.py b/python/python/ci_benchmarks/datagen/wikipedia.py index b64b60fea47..b08a5943634 100644 --- a/python/python/ci_benchmarks/datagen/wikipedia.py +++ b/python/python/ci_benchmarks/datagen/wikipedia.py @@ -133,7 +133,7 @@ def _create_indices(ds: lance.LanceDataset): Args: ds: Lance dataset to create indices on """ - existing_indices = [idx.name for idx in ds.list_indices()] + existing_indices = {idx.name for idx in ds.describe_indices()} # Create inverted index with position support for phrase queries # This index supports both match and phrase queries @@ -164,7 +164,7 @@ def _create(dataset_uri: str): LOGGER.info("Dataset exists with %s rows", f"{row_count:,}") # Check if indices exist - existing_indices = [idx.name for idx in ds.list_indices()] + existing_indices = {idx.name for idx in ds.describe_indices()} if "text_fts_idx" in existing_indices: LOGGER.info("Dataset and indices already exist, skipping generation") return