* Add warning for models not found in parser

This commit is contained in:
Matthew Honnibal 2015-07-08 19:52:30 +02:00
parent 78db7e32f7
commit f6a6c39ce8
1 changed files with 8 additions and 1 deletions

View File

@ -14,6 +14,7 @@ import os.path
from os import path from os import path
import shutil import shutil
import json import json
import sys
from cymem.cymem cimport Pool, Address from cymem.cymem cimport Pool, Address
from murmurhash.mrmr cimport hash64 from murmurhash.mrmr cimport hash64
@ -69,7 +70,13 @@ def ParserFactory(transition_system):
cdef class Parser: cdef class Parser:
def __init__(self, StringStore strings, model_dir, transition_system): def __init__(self, StringStore strings, model_dir, transition_system):
assert os.path.exists(model_dir) and os.path.isdir(model_dir) if not os.path.exists(model_dir):
print >> sys.stderr, "Warning: No model found at", model_dir
self.cfg = Config(labels=[], features=[])
elif not os.path.isdir(model_dir):
self.cfg = Config(labels=[], features=[])
print >> sys.stderr, "Warning: model path:", model_dir, "is not a directory"
self.cfg = Config.read(model_dir, 'config') self.cfg = Config.read(model_dir, 'config')
self.moves = transition_system(strings, self.cfg.labels) self.moves = transition_system(strings, self.cfg.labels)
templates = get_templates(self.cfg.features) templates = get_templates(self.cfg.features)