Skip to content

[BUG] cuvs_bench export fails if index was not performed #2025

@KyleFromNVIDIA

Description

@KyleFromNVIDIA

Describe the bug
When running the benchmarks with search but not build, and then exporting them, the following error is thrown:

Error processing build file /raid/kyedwards/benchmarks/main/datasets/deep-image-96-inner/result/build/cuvs_ivf_flat,base.json: 'name'. Skipping...
Traceback (most recent call last):
  File "/raid/kyedwards/miniforge3/envs/cuvs_benchmarks_main/lib/python3.14/site-packages/cuvs_bench/run/data_export.py", line 157, in convert_json_to_csv_build
    write_csv(file, algo_name, df, skip_cols=skip_build_cols)
    ~~~~~~~~~^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/raid/kyedwards/miniforge3/envs/cuvs_benchmarks_main/lib/python3.14/site-packages/cuvs_bench/run/data_export.py", line 123, in write_csv
    df["name"] = df["name"].str.split("/").str[0]
                 ~~^^^^^^^^
  File "/raid/kyedwards/miniforge3/envs/cuvs_benchmarks_main/lib/python3.14/site-packages/pandas/core/frame.py", line 4378, in __getitem__
    indexer = self.columns.get_loc(key)
  File "/raid/kyedwards/miniforge3/envs/cuvs_benchmarks_main/lib/python3.14/site-packages/pandas/core/indexes/range.py", line 525, in get_loc
    raise KeyError(key)
KeyError: 'name'

Adding the --search flag when exporting does not help.

Steps/Code to reproduce bug

  1. Run the benchmarks with search but not build.
  2. Follow the recommended instructions to export the results.

Expected behavior
The search results should be exported gracefully without the build results.

Environment details (please complete the following information):

  • Environment location: [Bare-metal, Docker, Cloud(specify cloud provider)]
  • Method of RAFT install: [conda, Docker, or from source]
    • If method of install is [Docker], provide docker pull & docker run commands used

Additional context
Add any other context about the problem here.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    Status

    Todo

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions