From a9b369681510f79780846b7447d73c342c48f394 Mon Sep 17 00:00:00 2001
From: George Stykalin <gstykalin@mail.ru>
Date: Tue, 14 Apr 2026 23:56:26 +0300
Subject: [PATCH] update dags

---
 dags/test-train-pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dags/test-train-pytorch.py b/dags/test-train-pytorch.py
index 4e1f869..720a87f 100644
--- a/dags/test-train-pytorch.py
+++ b/dags/test-train-pytorch.py
@@ -83,7 +83,7 @@ def run_training_node_func(rank, world_size):
   os.environ['MASTER_PORT'] = MASTER_PORT
   os.environ['WORLD_SIZE'] = str(world_size)
   os.environ['RANK'] = str(rank)
-  # os.environ['NCCL_SOCKET_IFNAME'] = 'eth0'
+  os.environ['NCCL_SOCKET_IFNAME'] = 'eth0'
   os.environ['NCCL_DEBUG'] = 'INFO'
   os.environ['NCCL_TIMEOUT'] = str(NCCL_TIMEOUT)
   os.environ['NCCL_BLOCKING_WAIT'] = '1'