Fixing illogical types in alignmentInfo and detokenize

ufal · Jul 10, 2014 · d4b312c · d4b312c
1 parent aa6bd3b
commit d4b312c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/appserver/src/appserver.py b/appserver/src/appserver.py
@@ -121,8 +121,8 @@ def _validate(self, task):
                 "targetLang": {"type": "string"},
                 "text": {"type": "string"},
                 "nBestSize": {"type": "integer", "required": False},
-                "detokenize": {"type": "integer", "required": False},
-                "alignmentInfo": {"type": "string", "required": False},
+                "detokenize": {"type": ['boolean', 'string', 'integer'], "required": False},
+                "alignmentInfo": {"type": ['boolean', 'string', 'integer'], "required": False},
                 "docType": {"type": "string", "required": False},
                 "profileType": {"type": "string", "required": False},
             },

diff --git a/worker/src/tasks/translate.py b/worker/src/tasks/translate.py
@@ -31,8 +31,8 @@ def __init__(self, translate_port, recase_port, source_lang, target_lang):
     def process_task(self, task):
         """Process translation task. Splits request into sentences, then translates and
         recases each sentence."""
-        doalign = task.get('alignmentInfo', '').lower() in ['true', 't', 'yes', 'y', '1']
-        dodetok = not task.get('detokenize', '').lower() in ['false', 'f', 'no', 'n', '0']
+        doalign = unicode(task.get('alignmentInfo', '')).lower() in ['true', 't', 'yes', 'y', '1']
+        dodetok = not unicode(task.get('detokenize', '')).lower() in ['false', 'f', 'no', 'n', '0']
         nbestsize = min(task.get('nBestSize', 1), 10)
         src_lines = self.splitter.split_sentences(task['text'])
         translated = [self._translate(line, doalign, dodetok, nbestsize) for line in src_lines]