mirror of
https://github.com/explosion/spaCy.git
synced 2024-12-24 17:06:29 +03:00
Try to fix tests on Travis (2.7)
This commit is contained in:
parent
f2a131bd9a
commit
56de520afd
|
@ -1,3 +1,4 @@
|
|||
# coding: utf8
|
||||
from __future__ import unicode_literals
|
||||
|
||||
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# coding: utf8
|
||||
from __future__ import unicode_literals
|
||||
|
||||
from ...attrs import LIKE_NUM
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# coding: utf8
|
||||
from __future__ import unicode_literals
|
||||
|
||||
|
||||
|
|
|
@ -109,6 +109,7 @@ class ChineseTokenizer(DummyTokenizer):
|
|||
if reset:
|
||||
try:
|
||||
import pkuseg
|
||||
|
||||
self.pkuseg_seg.preprocesser = pkuseg.Preprocesser(None)
|
||||
except ImportError:
|
||||
if self.use_pkuseg:
|
||||
|
@ -118,7 +119,7 @@ class ChineseTokenizer(DummyTokenizer):
|
|||
)
|
||||
raise ImportError(msg)
|
||||
for word in words:
|
||||
self.pkuseg_seg.preprocesser.insert(word.strip(), '')
|
||||
self.pkuseg_seg.preprocesser.insert(word.strip(), "")
|
||||
|
||||
def _get_config(self):
|
||||
config = OrderedDict(
|
||||
|
@ -168,21 +169,19 @@ class ChineseTokenizer(DummyTokenizer):
|
|||
return util.to_bytes(serializers, [])
|
||||
|
||||
def from_bytes(self, data, **kwargs):
|
||||
pkuseg_features_b = b""
|
||||
pkuseg_weights_b = b""
|
||||
pkuseg_processors_data = None
|
||||
data = {"features_b": b"", "weights_b": b"", "processors_data": None}
|
||||
# pkuseg_features_b = b""
|
||||
# pkuseg_weights_b = b""
|
||||
# pkuseg_processors_data = None
|
||||
|
||||
def deserialize_pkuseg_features(b):
|
||||
nonlocal pkuseg_features_b
|
||||
pkuseg_features_b = b
|
||||
data["features_b"] = b
|
||||
|
||||
def deserialize_pkuseg_weights(b):
|
||||
nonlocal pkuseg_weights_b
|
||||
pkuseg_weights_b = b
|
||||
data["weights_b"] = b
|
||||
|
||||
def deserialize_pkuseg_processors(b):
|
||||
nonlocal pkuseg_processors_data
|
||||
pkuseg_processors_data = srsly.msgpack_loads(b)
|
||||
data["processors_data"] = srsly.msgpack_loads(b)
|
||||
|
||||
deserializers = OrderedDict(
|
||||
(
|
||||
|
@ -194,13 +193,13 @@ class ChineseTokenizer(DummyTokenizer):
|
|||
)
|
||||
util.from_bytes(data, deserializers, [])
|
||||
|
||||
if pkuseg_features_b and pkuseg_weights_b:
|
||||
if data["features_b"] and data["weights_b"]:
|
||||
with tempfile.TemporaryDirectory() as tempdir:
|
||||
tempdir = Path(tempdir)
|
||||
with open(tempdir / "features.pkl", "wb") as fileh:
|
||||
fileh.write(pkuseg_features_b)
|
||||
fileh.write(data["features_b"])
|
||||
with open(tempdir / "weights.npz", "wb") as fileh:
|
||||
fileh.write(pkuseg_weights_b)
|
||||
fileh.write(data["weights_b"])
|
||||
try:
|
||||
import pkuseg
|
||||
except ImportError:
|
||||
|
@ -209,13 +208,10 @@ class ChineseTokenizer(DummyTokenizer):
|
|||
+ _PKUSEG_INSTALL_MSG
|
||||
)
|
||||
self.pkuseg_seg = pkuseg.pkuseg(str(tempdir))
|
||||
if pkuseg_processors_data:
|
||||
(
|
||||
user_dict,
|
||||
do_process,
|
||||
common_words,
|
||||
other_words,
|
||||
) = pkuseg_processors_data
|
||||
if data["processors_data"]:
|
||||
(user_dict, do_process, common_words, other_words) = data[
|
||||
"processors_data"
|
||||
]
|
||||
self.pkuseg_seg.preprocesser = pkuseg.Preprocesser(user_dict)
|
||||
self.pkuseg_seg.postprocesser.do_process = do_process
|
||||
self.pkuseg_seg.postprocesser.common_words = set(common_words)
|
||||
|
|
|
@ -1,3 +1,4 @@
|
|||
# coding: utf8
|
||||
from __future__ import unicode_literals
|
||||
|
||||
import pytest
|
||||
|
|
Loading…
Reference in New Issue
Block a user