From 9271bddf511a5ba1608c7a6efddc4cbe82659378 Mon Sep 17 00:00:00 2001 From: Dimo Angelov Date: Tue, 10 Nov 2020 11:15:59 -0500 Subject: [PATCH] update requirement error messages --- top2vec/Top2Vec.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/top2vec/Top2Vec.py b/top2vec/Top2Vec.py index c23a36d..8fe676b 100644 --- a/top2vec/Top2Vec.py +++ b/top2vec/Top2Vec.py @@ -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: