Adding BASE and NMT constants for translation.

googleapis · Nov 14, 2016 · de69ded · de69ded
1 parent d7175b0
commit de69ded
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/packages/google-cloud-translate/google/cloud/translate/__init__.py b/packages/google-cloud-translate/google/cloud/translate/__init__.py
@@ -14,5 +14,7 @@
 
 """Google Cloud Translate API wrapper."""
 
+from google.cloud.translate.client import BASE
 from google.cloud.translate.client import Client
+from google.cloud.translate.client import NMT
 from google.cloud.translate.connection import Connection
diff --git a/packages/google-cloud-translate/google/cloud/translate/client.py b/packages/google-cloud-translate/google/cloud/translate/client.py
@@ -26,6 +26,12 @@
 ENGLISH_ISO_639 = 'en'
 """ISO 639-1 language code for English."""
 
+BASE = 'base'
+"""Base translation model."""
+
+NMT = 'nmt'
+"""Neural Machine Translation model."""
+
 
 class Client(BaseClient):
     """Client to bundle configuration needed for API requests.
@@ -194,7 +200,7 @@ def translate(self, values, target_language=None, format_=None,
 
         :type model: str
         :param model: (Optional) The model used to translate the text. The
-                      only accepted values are ``base`` and ``nmt``.
+                      only accepted values are :attr:`BASE` and :attr:`NMT`.
 
         :rtype: str or list
         :returns: A list of dictionaries for each queried value. Each