Skip to content

Commit

Permalink
Updated exception handling for str type (#160)
Browse files Browse the repository at this point in the history
  • Loading branch information
benhoff authored and davisking committed Nov 7, 2017
1 parent a418400 commit 0889996
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions mitielib/mitie.py
Original file line number Diff line number Diff line change
Expand Up @@ -179,7 +179,7 @@ def load_entire_file(filename):
filename = to_bytes(filename)
x = _f.mitie_load_entire_file(filename)
if x is None:
raise Exception("Unable to load file " + filename)
raise Exception("Unable to load file " + to_default_str_type(filename))
res = ctypes.string_at(x)
_f.mitie_free(x)
return res
Expand Down Expand Up @@ -269,10 +269,10 @@ def save_to_disk(self, filename, pure_model=False):
filename = to_bytes(filename)
if (pure_model):
if (_f.mitie_save_named_entity_extractor_pure_model(filename, self.__obj) != 0):
raise Exception("Unable to save named_entity_extractor to the file " + filename);
raise Exception("Unable to save named_entity_extractor to the file " + to_default_str_type(filename));
else:
if (_f.mitie_save_named_entity_extractor(filename, self.__obj) != 0):
raise Exception("Unable to save named_entity_extractor to the file " + filename);
raise Exception("Unable to save named_entity_extractor to the file " + to_default_str_type(filename));

def extract_entities(self, tokens, feature_extractor=None):
tags = self.get_possible_ner_tags()
Expand Down Expand Up @@ -730,7 +730,7 @@ def save_to_disk(self, filename,pure_model=False):
filename = to_bytes(filename)
if (pure_model):
if (_f.mitie_save_text_categorizer_pure_model(filename, self.__obj) != 0):
raise Exception("Unable to save text_categorizer to the file " + filename);
raise Exception("Unable to save text_categorizer to the file " + to_default_str_type(filename));
else:
if (_f.mitie_save_text_categorizer(filename, self.__obj) != 0):
raise Exception("Unable to save text_categorizer to the file " + to_default_str_type(filename));
Expand Down Expand Up @@ -762,7 +762,7 @@ def __init__(self, filename):
self.__obj = _f.mitie_create_text_categorizer_trainer(filename)
self.__mitie_free = _f.mitie_free
if self.__obj is None:
raise Exception("Unable to create text_categorizer_trainer based on " + filename)
raise Exception("Unable to create text_categorizer_trainer based on " + to_default_str_type(filename))

def __del__(self):
self.__mitie_free(self.__obj)
Expand Down Expand Up @@ -827,7 +827,7 @@ def __init__(self, filename):
else:
self.__obj = _f.mitie_load_total_word_feature_extractor(filename)
if self.__obj is None:
raise Exception("Unable to load total_word_feature_extractor detector from " + filename)
raise Exception("Unable to load total_word_feature_extractor detector from " + to_default_str_type(filename))

def __del__(self):
self.__mitie_free(self.__obj)
Expand Down

0 comments on commit 0889996

Please sign in to comment.