Import sys.exit() to raise error if gensim not available

This error-handling may not comply with the general litstudy convention.
NLeSC · Jan 5, 2024 · bcec9c0 · bcec9c0
1 parent 98b7eb5
commit bcec9c0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/litstudy/nlp.py b/litstudy/nlp.py
@@ -328,7 +328,8 @@ def train_lda_model(corpus: Corpus, num_topics, seed=0, **kwargs) -> TopicModel:
         from gensim.models.ldamodel import LdaModel
         model = LdaModel(freqs, id2word=dic, **kwargs)
     else:
-        sys.exit('LdaModel could not be imported from gensim 3 or 4.')
+        from sys import exit
+        exit('LdaModel could not be imported from gensim 3 or 4.')
 
     doc2topic = corpus2dense(model[freqs], num_topics)
     topic2token = model.get_topics()