(airflow) branch main updated: Fix translate gcp native model system test (#51238)

[shahar]

This is an automated email from the ASF dual-hosted git repository.

shahar pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new d3fb4c1d941 Fix translate gcp native model system test (#51238)
d3fb4c1d941 is described below

commit d3fb4c1d9418fd9329b99285e34a7888a17ef037
Author: olegkachur-e <qws...@gmail.com>
AuthorDate: Sat May 31 11:01:09 2025 +0200

    Fix translate gcp native model system test (#51238)
    
    Co-authored-by: Oleg Kachur <kac...@google.com>
---
 .../google/cloud/translate/example_translate_model.py | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git 
a/providers/google/tests/system/google/cloud/translate/example_translate_model.py
 
b/providers/google/tests/system/google/cloud/translate/example_translate_model.py
index 3514668fa2d..6a6d919071b 100644
--- 
a/providers/google/tests/system/google/cloud/translate/example_translate_model.py
+++ 
b/providers/google/tests/system/google/cloud/translate/example_translate_model.py
@@ -45,26 +45,26 @@ ENV_ID = os.environ.get("SYSTEM_TESTS_ENV_ID", "default")
 REGION = "us-central1"
 RESOURCE_DATA_BUCKET = "airflow-system-tests-resources"
 DATA_SAMPLE_GCS_BUCKET_NAME = f"bucket_{DAG_ID}_{ENV_ID}".replace("_", "-")
-DATA_FILE_NAME = "import_en-es_short.tsv"
+DATA_FILE_NAME = "import_en-es.tsv"
 RESOURCE_PATH = f"V3_translate/create_ds/import_data/{DATA_FILE_NAME}"
 COPY_DATA_PATH = 
f"gs://{RESOURCE_DATA_BUCKET}/V3_translate/create_ds/import_data/{DATA_FILE_NAME}"
 DST_PATH = f"translate/import/{DATA_FILE_NAME}"
 DATASET_DATA_PATH = f"gs://{DATA_SAMPLE_GCS_BUCKET_NAME}/{DST_PATH}"
 DATASET = {
-    "display_name": f"op_ds_native{DAG_ID}_{ENV_ID}",
-    "source_language_code": "es",
-    "target_language_code": "en",
+    "display_name": f"ds_native_{DAG_ID}_{ENV_ID}",
+    "source_language_code": "en",
+    "target_language_code": "es",
 }
 
 
 with DAG(
     DAG_ID,
     schedule="@once",  # Override to match your needs
-    start_date=datetime(2024, 11, 1),
+    start_date=datetime(2025, 1, 1),
     catchup=False,
     tags=[
         "example",
-        "translate_model",
+        "translate_native_model",
     ],
 ) as dag:
     create_bucket = GCSCreateBucketOperator(
@@ -120,11 +120,12 @@ with DAG(
 
     translate_text_with_model = TranslateTextOperator(
         task_id="translate_v3_op",
-        contents=["Hola!", "Puedes traerme una taza de café, por favor?"],
+        contents=["Hello!", "Can I have a cup of coffee, please?"],
         # AutoML model format
         model=f"projects/{PROJECT_ID}/locations/{REGION}/models/{model_id}",
-        source_language_code="es",
-        target_language_code="en",
+        source_language_code="en",
+        target_language_code="es",
+        location=REGION,
     )
 
     # [START howto_operator_translate_automl_delete_model]