File tree 3 files changed +3
-4
lines changed
3 files changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -161,7 +161,6 @@ def create_collection(
161
161
get_or_create = get_or_create ,
162
162
configuration = configuration ,
163
163
)
164
- print (f"(Sanket-temp) Collection model { model .configuration_json } " )
165
164
return Collection (
166
165
client = self ._server ,
167
166
model = model ,
Original file line number Diff line number Diff line change @@ -246,11 +246,12 @@ def fd_not_exceeding_threadpool_size(threadpool_size: int) -> None:
246
246
)
247
247
248
248
def get_space (collection : Collection ):
249
- print (f"(Sanket-temp) Collection config json { collection ._model .configuration_json } " )
250
249
if 'spann' in collection ._model .configuration_json and collection ._model .configuration_json .get ('spann' ) is not None :
251
250
return collection ._model .configuration_json .get ('spann' ).get ('space' )
252
- else :
251
+ elif 'hnsw' in collection . _model . configuration_json and collection . _model . configuration_json . get ( 'hnsw' ) is not None :
253
252
return collection ._model .configuration_json .get ('hnsw' ).get ('space' )
253
+ else :
254
+ return collection .metadata ["hnsw:space" ]
254
255
255
256
def ann_accuracy (
256
257
collection : Collection ,
Original file line number Diff line number Diff line change @@ -127,7 +127,6 @@ def test_persist(
127
127
metadata = collection_strategy .metadata , # type: ignore[arg-type]
128
128
embedding_function = collection_strategy .embedding_function ,
129
129
)
130
- print (f"(Sanket-temp) Collection config { coll .configuration } " )
131
130
132
131
coll .add (** recordset_strategy ) # type: ignore[arg-type]
133
132
You can’t perform that action at this time.
0 commit comments