Skip to content

Commit 3c8dc41

Browse files
satymishfacebook-github-bot
authored andcommitted
Set KnnDescriptor.desc_name in the Benchmarking core framework in FAISS like other descriptors (facebookresearch#4109)
Summary: Pull Request resolved: facebookresearch#4109 Set KnnDescriptor.desc_name in the Benchmarking core framework in FAISS like other descriptors Reviewed By: mnorris11 Differential Revision: D67539874 fbshipit-source-id: 09ffb76296f466ae2d3b0eb551917f429bc7300f
1 parent ab8cb9c commit 3c8dc41

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

benchs/bench_fw/descriptors.py

+3
Original file line numberDiff line numberDiff line change
@@ -341,6 +341,8 @@ def __hash__(self):
341341
return hash(str(self))
342342

343343
def get_name(self):
344+
if self.desc_name is not None:
345+
return self.desc_name
344346
name = self.index_desc.get_name()
345347
name += IndexBaseDescriptor.param_dict_to_name(self.search_params)
346348
name += self.query_dataset.get_filename(KnnDescriptor.FILENAME_PREFIX)
@@ -350,6 +352,7 @@ def get_name(self):
350352
name += "rec."
351353
else:
352354
name += "knn."
355+
self.desc_name = name
353356
return name
354357

355358
def flat_name(self):

0 commit comments

Comments
 (0)