Skip to content

Commit

Permalink
upgrade spacy to the latest (#204)
Browse files Browse the repository at this point in the history
  • Loading branch information
jackjii79 authored Jun 22, 2024
1 parent deed860 commit 6394ca2
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions transformers/nlp/text_named_entities_transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,8 @@ class TextNamedEntityTransformer(CustomTransformer):
'%s/wasabi-0.8.2%s' % (_root_path, _suffix),
]
else:
_modules_needed_by_name = ["spacy==3.7.4",
"https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-2.2.5/en_core_web_sm-2.2.5.tar.gz#egg=en_core_web_sm==2.2.5"]
_modules_needed_by_name = ["spacy==3.7.5",
"https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.7.0/en_core_web_sm-3.7.0.tar.gz#egg=en_core_web_sm==3.7.0"]

def __init__(self, **kwargs):
super().__init__(**kwargs)
Expand All @@ -57,7 +57,6 @@ def fit_transform(self, X: dt.Frame, y: np.array = None):
return self.transform(X)

def transform(self, X: dt.Frame):
import spacy
import en_core_web_sm
nlp = en_core_web_sm.load()

Expand Down

0 comments on commit 6394ca2

Please sign in to comment.