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

View File

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