From 5fb958cc83f9abc36d0fa06a443cd2b0b21b3bc6 Mon Sep 17 00:00:00 2001
From: "Gao, Ruiyuan" <905370712@qq.com>
Date: Wed, 24 Jul 2024 10:30:40 +0800
Subject: [PATCH] [FIX BUG] convert env param to int in (#5934)

---
 colossalai/cluster/dist_coordinator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/cluster/dist_coordinator.py b/colossalai/cluster/dist_coordinator.py
index 98191747e..14a8eabb4 100644
--- a/colossalai/cluster/dist_coordinator.py
+++ b/colossalai/cluster/dist_coordinator.py
@@ -44,7 +44,7 @@ class DistCoordinator(metaclass=SingletonMeta):
         self._rank = dist.get_rank()
         self._world_size = dist.get_world_size()
         # this is often passed by launchers such as torchrun
-        self._local_rank = os.environ.get("LOCAL_RANK", -1)
+        self._local_rank = int(os.environ.get("LOCAL_RANK", -1))
 
     @property
     def rank(self) -> int: