update gensim to code inside.

leonnewton · Sep 16, 2023 · 63be6d2 · 63be6d2
1 parent d4b4d79
commit 63be6d2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,5 +4,4 @@ transformers>=4.6.0
 datasets
 tqdm
 scikit-learn
-gensim>=4.0.0
 pandas
diff --git a/setup.py b/setup.py
@@ -44,7 +44,6 @@
         "datasets",
         "tqdm",
         "scikit-learn",
-        "gensim>=4.0.0",
         "pandas",
     ],
     packages=find_packages(exclude=['tests']),

diff --git a/text2vec/word2vec.py b/text2vec/word2vec.py
@@ -9,7 +9,6 @@
 from typing import List, Union, Dict
 
 import numpy as np
-from gensim.models import KeyedVectors
 from loguru import logger
 from numpy import ndarray
 from tqdm import tqdm
@@ -63,6 +62,8 @@ def __init__(self, model_name_or_path: str = 'w2v-light-tencent-chinese',
             stopwords: list, stopwords
             cache_folder: str, save model dir
         """
+        from gensim.models import KeyedVectors  # noqa
+
         self.w2v_kwargs = w2v_kwargs if w2v_kwargs is not None else {}
         if model_name_or_path and os.path.exists(model_name_or_path):
             logger.info('Load pretrained model:{}'.format(model_name_or_path))