@@ -130,11 +130,11 @@ class IndexConfig:
130
130
"""Configuration options for the Vertex FeatureView for embedding."""
131
131
132
132
embedding_column : str
133
- filter_column : List [str ]
134
- crowding_column : str
135
- dimentions : Optional [int ]
136
- distance_measure_type : DistanceMeasureType
133
+ dimensions : int
137
134
algorithm_config : AlgorithmConfig
135
+ filter_columns : Optional [List [str ]] = None
136
+ crowding_column : Optional [str ] = None
137
+ distance_measure_type : Optional [DistanceMeasureType ] = None
138
138
139
139
def as_dict (self ) -> Dict [str , Any ]:
140
140
"""Returns the configuration as a dictionary.
@@ -144,11 +144,15 @@ def as_dict(self) -> Dict[str, Any]:
144
144
"""
145
145
config = {
146
146
"embedding_column" : self .embedding_column ,
147
- "filter_columns" : self .filter_column ,
148
- "crowding_column" : self .crowding_column ,
149
- "embedding_dimension" : self .dimentions ,
150
- "distance_measure_type" : self .distance_measure_type .value ,
147
+ "embedding_dimension" : self .dimensions ,
151
148
}
149
+ if self .distance_measure_type is not None :
150
+ config ["distance_measure_type" ] = self .distance_measure_type .value
151
+ if self .filter_columns is not None :
152
+ config ["filter_columns" ] = self .filter_columns
153
+ if self .crowding_column is not None :
154
+ config ["crowding_column" ] = self .crowding_column
155
+
152
156
if isinstance (self .algorithm_config , TreeAhConfig ):
153
157
config ["tree_ah_config" ] = self .algorithm_config .as_dict ()
154
158
else :
0 commit comments