diff --git a/src/search_type/asymmetric.py b/src/search_type/asymmetric.py index b65d667f..5b55123b 100644 --- a/src/search_type/asymmetric.py +++ b/src/search_type/asymmetric.py @@ -59,7 +59,7 @@ def compute_embeddings(entries, bi_encoder, embeddings_file, regenerate=False, v corpus_embeddings = bi_encoder.encode([entry[0] for entry in entries], convert_to_tensor=True, show_progress_bar=True) torch.save(corpus_embeddings, get_absolute_path(embeddings_file)) if verbose > 0: - print(f"Computed embeddings and save them to {embeddings_file}") + print(f"Computed embeddings and saved them to {embeddings_file}") return corpus_embeddings diff --git a/src/search_type/symmetric_ledger.py b/src/search_type/symmetric_ledger.py index c507bbf5..d48665bd 100644 --- a/src/search_type/symmetric_ledger.py +++ b/src/search_type/symmetric_ledger.py @@ -53,7 +53,7 @@ def compute_embeddings(entries, bi_encoder, embeddings_file, regenerate=False, v corpus_embeddings = bi_encoder.encode(entries, convert_to_tensor=True, show_progress_bar=True) torch.save(corpus_embeddings, get_absolute_path(embeddings_file)) if verbose > 0: - print(f"Computed embeddings and save them to {embeddings_file}") + print(f"Computed embeddings and saved them to {embeddings_file}") return corpus_embeddings