Skip to content

Commit 0358422

Browse files
committed
fixing language translation models
1 parent b565f5d commit 0358422

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

examples/language_translation_v2.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,8 @@
2121
print(json.dumps(language_translation.translate('Hola, cómo estás? €', source='es', target='en'), indent=2,
2222
ensure_ascii=False))
2323

24-
print(json.dumps(language_translation.identify('Hello how are you?'), indent=2))
24+
print(json.dumps(language_translation.translate('Messi is the best ever', model_id='en-es-conversational'), indent=2))
2525

26+
print(json.dumps(language_translation.identify('Hello how are you?'), indent=2))
2627

2728
print(json.dumps(language_translation.get_identifiable_languages(), indent=2))

watson_developer_cloud/language_translation_v2.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -62,14 +62,14 @@ def get_model(self, model_id):
6262
def delete_model(self, model_id):
6363
return self.request(method='DELETE', url='/v2/models/{0}'.format(model_id), accept_json=True)
6464

65-
def translate(self, text, source=None, target=None, model=None):
65+
def translate(self, text, source=None, target=None, model_id=None):
6666
"""
6767
Translates text from a source language to a target language
6868
"""
69-
if model is None and (source is None or target is None):
70-
raise WatsonInvalidArgument('Either model or source and target must be specified')
69+
if model_id is None and (source is None or target is None):
70+
raise WatsonInvalidArgument('Either model_id or source and target must be specified')
7171

72-
data = {'text': text, 'source': source, 'target': target, 'model': model}
72+
data = {'text': text, 'source': source, 'target': target, 'model_id': model_id}
7373

7474
# data=data or json=data
7575
return self.request(method='POST', url='/v2/translate', json=data).text

0 commit comments

Comments
 (0)