move except

Author: manuelporrasojeda

toolium/utils/ai_utils/spacy.py

@@ -40,6 +40,8 @@ def get_spacy_model(model_name):
     :param model_name: spaCy model name
     :return: spaCy model
     """
+    if spacy is None:
+        raise ImportError("spaCy is not installed. Please run 'pip install toolium[ai]' to use spaCy features")
     config = DriverWrappersPool.get_default_wrapper().config
     model_name = model_name or config.get_optional('AI', 'spacy_model', 'en_core_web_md')
     return spacy.load(model_name)

toolium/utils/ai_utils/text_similarity.py

@@ -52,8 +52,6 @@ def get_text_similarity_with_spacy(text, expected_text, model_name=None):
     # - Normalizing texts (lowercase, extra points, etc.)
     # - Use only models that include word vectors (e.g., 'en_core_news_md' or 'en_core_news_lg')
     # - Preprocessing texts. Now we only preprocess negations.
-    if spacy is None:
-        raise ImportError("spaCy is not installed. Please run 'pip install toolium[ai]' to use spaCy features")
     model = get_spacy_model(model_name)
     text = model(preprocess_with_ud_negation(text, model))
     expected_text = model(preprocess_with_ud_negation(expected_text, model))