Skip to content

Commit

Permalink
update requirement error messages
Browse files Browse the repository at this point in the history
  • Loading branch information
ddangelov committed Nov 10, 2020
1 parent 55b6174 commit 9271bdd
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions top2vec/Top2Vec.py
Original file line number Diff line number Diff line change
Expand Up @@ -642,13 +642,13 @@ def _check_import_status(self):
if self.embedding_model != 'distiluse-base-multilingual-cased':
if not _HAVE_TENSORFLOW:
raise ImportError(f"{self.embedding_model} is not available.\n\n"
"Try pip install top2vec[sentence_encoders]\n\n"
"Alternatively try pip install tensorflow tensorflow_hub tensorflow_text")
"Try: pip install top2vec[sentence_encoders]\n\n"
"Alternatively try: pip install tensorflow tensorflow_hub tensorflow_text")
else:
if not _HAVE_TORCH:
raise ImportError(f"{self.embedding_model} is not available.\n\n"
"Try pip install top2vec[sentence_transformers]\n\n"
"Alternatively try pip install torch sentence_transformers")
"Try: pip install top2vec[sentence_transformers]\n\n"
"Alternatively try: pip install torch sentence_transformers")

def _check_model_status(self):
if self.embed is None:
Expand Down

0 comments on commit 9271bdd

Please sign in to comment.