Update imports

This commit is contained in:
Matthew Honnibal 2020-06-20 22:21:08 +02:00
parent cfd024536d
commit 64d00520e2
2 changed files with 4 additions and 4 deletions

View File

@ -4,7 +4,7 @@ import sys
import srsly import srsly
from wasabi import Printer, MESSAGES from wasabi import Printer, MESSAGES
from ..gold import GoldCorpus from ..gold import Corpus
from ..syntax import nonproj from ..syntax import nonproj
from ..util import load_model, get_lang_class from ..util import load_model, get_lang_class
@ -68,7 +68,7 @@ def debug_data(
loading_train_error_message = "" loading_train_error_message = ""
loading_dev_error_message = "" loading_dev_error_message = ""
with msg.loading("Loading corpus..."): with msg.loading("Loading corpus..."):
corpus = GoldCorpus(train_path, dev_path) corpus = Corpus(train_path, dev_path)
try: try:
train_dataset = list(corpus.train_dataset(nlp)) train_dataset = list(corpus.train_dataset(nlp))
train_dataset_unpreprocessed = list( train_dataset_unpreprocessed = list(

View File

@ -1,7 +1,7 @@
from timeit import default_timer as timer from timeit import default_timer as timer
from wasabi import msg from wasabi import msg
from ..gold import GoldCorpus from ..gold import Corpus
from .. import util from .. import util
from .. import displacy from .. import displacy
@ -31,7 +31,7 @@ def evaluate(
msg.fail("Evaluation data not found", data_path, exits=1) msg.fail("Evaluation data not found", data_path, exits=1)
if displacy_path and not displacy_path.exists(): if displacy_path and not displacy_path.exists():
msg.fail("Visualization output directory not found", displacy_path, exits=1) msg.fail("Visualization output directory not found", displacy_path, exits=1)
corpus = GoldCorpus(data_path, data_path) corpus = Corpus(data_path, data_path)
if model.startswith("blank:"): if model.startswith("blank:"):
nlp = util.get_lang_class(model.replace("blank:", ""))() nlp = util.get_lang_class(model.replace("blank:", ""))()
else: else: