Skip to content

Commit 44caa55

Browse files
authored
update test owner in inference (GoogleCloudPlatform#1124)
1 parent 68cd6ee commit 44caa55

3 files changed

Lines changed: 4 additions & 4 deletions

File tree

dags/inference/configs/jetstream_benchmark_serving_gce_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,7 @@ def get_config(
224224
set_up_cmds=set_up_cmds,
225225
run_model_cmds=run_model_cmds,
226226
timeout=datetime.timedelta(minutes=time_out_in_min),
227-
task_owner=test_owner.AIRFLOW,
227+
task_owner=test_owner.XIANG_S,
228228
num_slices=num_slices,
229229
gcs_subfolder=f"{GCS_SUBFOLDER_PREFIX}/maxtext",
230230
)

dags/inference/configs/maxtext_inference_microbenchmark_gce_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ def config(
131131
set_up_cmds=set_up_cmds,
132132
run_model_cmds=run_model_cmds,
133133
timeout=datetime.timedelta(minutes=time_out_in_min),
134-
task_owner=test_owner.AIRFLOW,
134+
task_owner=test_owner.XIANG_S,
135135
num_slices=num_slices,
136136
gcs_subfolder=f"{GCS_SUBFOLDER_PREFIX}/maxtext",
137137
)

dags/multipod/maxtext_gpu_end_to_end.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ def run_maxtext_tests(dag: models.DAG):
207207
num_slices=nnodes,
208208
cluster=XpkClusters.GPU_A3_CLUSTER,
209209
docker_image=DockerImage.MAXTEXT_GPU_JAX_STABLE.value,
210-
test_owner=test_owner.ANISHA_M,
210+
test_owner=test_owner.DORA_H,
211211
).run_with_quarantine(quarantine_task_group)
212212
stable_a3plus_gpu = gke_config.get_maxtext_end_to_end_gpu_gke_test_config(
213213
time_out_in_min=300,
@@ -216,7 +216,7 @@ def run_maxtext_tests(dag: models.DAG):
216216
num_slices=nnodes,
217217
cluster=XpkClusters.GPU_A3PLUS_CLUSTER,
218218
docker_image=DockerImage.MAXTEXT_GPU_JAX_STABLE.value,
219-
test_owner=test_owner.ANISHA_M,
219+
test_owner=test_owner.DORA_H,
220220
).run_with_quarantine(quarantine_task_group)
221221
stable_a3_gpu >> stable_a3plus_gpu
222222

0 commit comments

Comments
 (0)