diff --git a/benchs/bench_fw/descriptors.py b/benchs/bench_fw/descriptors.py index f7c2e9e9a0..cf0ee958fe 100644 --- a/benchs/bench_fw/descriptors.py +++ b/benchs/bench_fw/descriptors.py @@ -113,6 +113,8 @@ class DatasetDescriptor: # desc_name desc_name: Optional[str] = None + filename_suffix: Optional[str] = None + normalize_L2: bool = False def __hash__(self): @@ -138,6 +140,8 @@ def get_filename( ).replace("=", "_").replace("/", "_") if self.num_vectors is not None: filename += f"_{self.num_vectors}" + if self.filename_suffix is not None: + filename += f"_{self.filename_suffix}" filename += "." self.desc_name = filename