Commit 07a345c 1 parent 61eaf19 commit 07a345c Copy full SHA for 07a345c
File tree 1 file changed +9
-1
lines changed
1 file changed +9
-1
lines changed Original file line number Diff line number Diff line change @@ -92,13 +92,19 @@ class DatasetDescriptor:
92
92
bucket : Optional [str ] = None
93
93
path : Optional [str ] = None
94
94
95
+ # desc_name
96
+ desc_name : Optional [str ] = None
97
+
95
98
def __hash__ (self ):
96
99
return hash (self .get_filename ())
97
100
98
101
def get_filename (
99
102
self ,
100
103
prefix : Optional [str ] = None ,
101
104
) -> str :
105
+ if self .desc_name is not None :
106
+ return self .desc_name
107
+
102
108
filename = ""
103
109
if prefix is not None :
104
110
filename += prefix + "_"
@@ -111,7 +117,9 @@ def get_filename(
111
117
if self .num_vectors is not None :
112
118
filename += f"_{ self .num_vectors } "
113
119
filename += "."
114
- return filename
120
+
121
+ self .desc_name = filename
122
+ return self .desc_name
115
123
116
124
def get_kmeans_filename (self , k ):
117
125
return f"{ self .get_filename ()} kmeans_{ k } ."
You can’t perform that action at this time.
0 commit comments