From 3daf48cc2af1b31dbc4ca5c3626920b8e337f0c3 Mon Sep 17 00:00:00 2001 From: George Wang Date: Fri, 11 Oct 2024 19:27:56 -0700 Subject: [PATCH] Fix total_rows (#3942) Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3942 Reverted splitting of partition string to align with previous behavior of finding partition intersection rather than the union. Also changed util dependency from laser to vector_search, which makes more sense. Reviewed By: kuarora Differential Revision: D64274531 fbshipit-source-id: 4614ae42d0fb534c9c9ce3314fd3c26a0c74d049 --- benchs/bench_fw/descriptors.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/benchs/bench_fw/descriptors.py b/benchs/bench_fw/descriptors.py index 54bd47bbbf..ab7d09b8f5 100644 --- a/benchs/bench_fw/descriptors.py +++ b/benchs/bench_fw/descriptors.py @@ -113,7 +113,9 @@ def get_filename( assert self.tablename is not None filename += self.tablename if self.partitions is not None: - filename += "_" + "_".join(self.partitions).replace("=", "_") + filename += "_" + "_".join( + self.partitions + ).replace("=", "_").replace("/", "_") if self.num_vectors is not None: filename += f"_{self.num_vectors}" filename += "."