fixed a bug, output object is different from expected when feature ty…

…pe is word
Kensuke-Mitsuzawa · Nov 6, 2017 · 50613f1 · 50613f1
1 parent 5f4e60f
commit 50613f1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/DocumentFeatureSelection/models.py b/DocumentFeatureSelection/models.py
@@ -409,9 +409,13 @@ def get_word(self, row_col_val_tuple:ROW_COL_VAL, vocabulary:Dict[int,str])->Uni
         vocab = vocabulary[row_col_val_tuple.col]
         try:
             feature_object = json.loads(vocab)
+            if len(feature_object)==1:
+                # When feature is word, the length is 1 #
+                feature_object = feature_object[0]
         except:
             feature_object = vocab
 
+
         return feature_object
 
     def initialize_cache_dict_object(self, cache_backend:str, file_name:str, path_cache_file=mkdtemp()):

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 """
 
 __author__ = 'kensuke-mi'
-__version__ = '1.4'
+__version__ = '1.4.1'
 
 import sys
 print('python version {}'.format(sys.version_info))