From 214a3255e57d2ca5a6ac0831508c690668e93f3d Mon Sep 17 00:00:00 2001 From: Tamas Bela Feher Date: Thu, 25 Apr 2024 23:35:53 +0200 Subject: [PATCH 1/3] Replace too long index file name with hash in ANN bench --- python/raft-ann-bench/src/raft-ann-bench/run/__main__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py index 52d536c2e8..84bf258082 100644 --- a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py +++ b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py @@ -553,10 +553,10 @@ def add_algo_group(group_list): index["build_param"], conf_file["dataset"]["dims"] ): continue - + index_filename = index_name if len(index_name) < 128 else str(hash(index_name)) index["name"] = index_name index["file"] = os.path.join( - args.dataset_path, args.dataset, "index", index_name + args.dataset_path, args.dataset, "index", index_filename ) index["search_params"] = [] all_search_params = itertools.product(*search_param_lists) From 4d4d31946b660436c618b193ef11a632b4e4617b Mon Sep 17 00:00:00 2001 From: Tamas Bela Feher Date: Tue, 21 May 2024 20:31:42 +0200 Subject: [PATCH 2/3] fix-styel --- python/raft-ann-bench/src/raft-ann-bench/run/__main__.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py index 84bf258082..31c20538be 100644 --- a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py +++ b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py @@ -1,5 +1,5 @@ # -# Copyright (c) 2023, NVIDIA CORPORATION. +# Copyright (c) 2023-2024, NVIDIA CORPORATION. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -553,7 +553,11 @@ def add_algo_group(group_list): index["build_param"], conf_file["dataset"]["dims"] ): continue - index_filename = index_name if len(index_name) < 128 else str(hash(index_name)) + index_filename = ( + index_name + if len(index_name) < 128 + else str(hash(index_name)) + ) index["name"] = index_name index["file"] = os.path.join( args.dataset_path, args.dataset, "index", index_filename From ad6731711048f8c19c04a4ea474db22219d75ce7 Mon Sep 17 00:00:00 2001 From: Tamas Bela Feher Date: Tue, 21 May 2024 20:35:55 +0200 Subject: [PATCH 3/3] fix style --- python/raft-ann-bench/src/raft-ann-bench/run/__main__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py index 31c20538be..c34377d733 100644 --- a/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py +++ b/python/raft-ann-bench/src/raft-ann-bench/run/__main__.py @@ -554,8 +554,8 @@ def add_algo_group(group_list): ): continue index_filename = ( - index_name - if len(index_name) < 128 + index_name + if len(index_name) < 128 else str(hash(index_name)) ) index["name"] = index_name