From 2edf537df68db8bc929be446b0af40a59553e330 Mon Sep 17 00:00:00 2001 From: George Stykalin Date: Mon, 26 Jan 2026 15:49:45 +0300 Subject: [PATCH] update dags --- dags/connection-check.py | 2 +- dags/test-train-pytorch.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dags/connection-check.py b/dags/connection-check.py index e5d4a63..520a11e 100644 --- a/dags/connection-check.py +++ b/dags/connection-check.py @@ -2,7 +2,7 @@ from airflow import DAG from airflow.decorators import task import pendulum -TARGET_HOST = "airflow-k8s-demo-0164-worker-gpu-0.airflow-k8s-demo-0164-worker-gpu" +TARGET_HOST = "airflow-k8s-demo-0168-worker-gpu-0.airflow-k8s-demo-0168-worker-gpu" with DAG( dag_id='debug_gpu_network_check', diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py index 94ba434..48e219e 100644 --- a/dags/test-train-pytorch.py +++ b/dags/test-train-pytorch.py @@ -7,7 +7,7 @@ import json # --- CONFIGURATION --- WORLD_SIZE = 2 -MASTER_ADDR = "airflow-k8s-demo-0167-worker-gpu-0.airflow-k8s-demo-0167-worker-gpu" +MASTER_ADDR = "airflow-k8s-demo-0168-worker-gpu-0.airflow-k8s-demo-0168-worker-gpu" MASTER_PORT = "29500" NCCL_TIMEOUT = 1800