explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-25 17:36:30 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix cytoolz import cytoolz

Browse Source
This commit is contained in:
Matthew Honnibal 2018-12-06 16:04:12 +01:00
parent cabaadd793
commit 711f108532
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/cli/profile.py
View File

@ -8,7 +8,7 @@ import cProfile
import pstats import pstats
import sys import sys
import tqdm import tqdm
import cytoolz import itertools
import thinc.extra.datasets import thinc.extra.datasets
from wasabi import Printer from wasabi import Printer
@ -40,7 +40,7 @@ def profile(model, inputs=None, n_texts=10000):
with msg.loading("Loading model '{}'...".format(model)): with msg.loading("Loading model '{}'...".format(model)):
nlp = load_model(model) nlp = load_model(model)
msg.good("Loaded model '{}'".format(model)) msg.good("Loaded model '{}'".format(model))
texts = list(cytoolz.take(n_texts, inputs)) texts = list(itertools.islice(inputs, n_texts))
cProfile.runctx("parse_texts(nlp, texts)", globals(), locals(), "Profile.prof") cProfile.runctx("parse_texts(nlp, texts)", globals(), locals(), "Profile.prof")
s = pstats.Stats("Profile.prof") s = pstats.Stats("Profile.prof")
msg.divider("Profile stats") msg.divider("Profile stats")