remove translate prediction fallback

translate/automl/automl_translation_predict.py

@@ -25,20 +25,13 @@
 import os
 
 
-def predict(
- project_id,
- compute_region,
- model_id,
- file_path,
- translation_allow_fallback=False,
-):
+def predict(project_id, compute_region, model_id, file_path):
  """Translate the content."""
  # [START automl_translation_predict]
  # project_id = 'PROJECT_ID_HERE'
  # compute_region = 'COMPUTE_REGION_HERE'
  # model_id = 'MODEL_ID_HERE'
  # file_path = '/local/path/to/file'
- # translation_allow_fallback = True allows fallback to Google Translate
 
  from google.cloud import automl_v1beta1 as automl
 
@@ -61,10 +54,7 @@ def predict(
  payload = {"text_snippet": {"content": content}}
 
  # params is additional domain-specific parameters.
- # translation_allow_fallback allows to use Google translation model.
  params = {}
- if translation_allow_fallback:
- params = {"translation_allow_fallback": "True"}
 
  response = prediction_client.predict(model_full_id, payload, params)
  translated_content = response.payload[0].translation.translated_content
@@ -84,26 +74,11 @@ def predict(
  predict_parser = subparsers.add_parser("predict", help=predict.__doc__)
  predict_parser.add_argument("model_id")
  predict_parser.add_argument("file_path")
- predict_parser.add_argument(
- "translation_allow_fallback",
- nargs="?",
- choices=["False", "True"],
- default="False",
- )
 
  project_id = os.environ["PROJECT_ID"]
  compute_region = os.environ["REGION_NAME"]
 
  args = parser.parse_args()
 
  if args.command == "predict":
- translation_allow_fallback = (
- True if args.translation_allow_fallback == "True" else False
- )
- predict(
- project_id,
- compute_region,
- args.model_id,
- args.file_path,
- translation_allow_fallback,
- )
+ predict(project_id, compute_region, args.model_id, args.file_path)

translate/automl/predict_test.py

@@ -25,7 +25,7 @@
 def test_predict(capsys):
  model_id = "3128559826197068699"
  automl_translation_predict.predict(
- project_id, compute_region, model_id, "resources/input.txt", False
+ project_id, compute_region, model_id, "resources/input.txt"
  )
  out, _ = capsys.readouterr()
  assert "Translated content: " in out