Skip to content

Commit 07a345c

Browse files
kuarorafacebook-github-bot
authored andcommitted
Add desc_name to dataset descriptor (#3935)
Summary: Pull Request resolved: #3935 same as title Reviewed By: satymish Differential Revision: D64144800 fbshipit-source-id: 4a298aa83315d82f44ee424bf0a30737d5bf48a4
1 parent 61eaf19 commit 07a345c

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

benchs/bench_fw/descriptors.py

+9-1
Original file line numberDiff line numberDiff line change
@@ -92,13 +92,19 @@ class DatasetDescriptor:
9292
bucket: Optional[str] = None
9393
path: Optional[str] = None
9494

95+
# desc_name
96+
desc_name: Optional[str] = None
97+
9598
def __hash__(self):
9699
return hash(self.get_filename())
97100

98101
def get_filename(
99102
self,
100103
prefix: Optional[str] = None,
101104
) -> str:
105+
if self.desc_name is not None:
106+
return self.desc_name
107+
102108
filename = ""
103109
if prefix is not None:
104110
filename += prefix + "_"
@@ -111,7 +117,9 @@ def get_filename(
111117
if self.num_vectors is not None:
112118
filename += f"_{self.num_vectors}"
113119
filename += "."
114-
return filename
120+
121+
self.desc_name = filename
122+
return self.desc_name
115123

116124
def get_kmeans_filename(self, k):
117125
return f"{self.get_filename()}kmeans_{k}."

0 commit comments

Comments
 (0)