From c21c3b5cc92c454c9e22399b8519f41e2a657828 Mon Sep 17 00:00:00 2001
From: flymin <905370712@qq.com>
Date: Tue, 23 Jul 2024 17:08:21 +0800
Subject: [PATCH] [FIX BUG] convert env param to int in

---
 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 98191747e5b3..14a8eabb42b1 100644
--- a/colossalai/cluster/dist_coordinator.py
+++ b/colossalai/cluster/dist_coordinator.py
@@ -44,7 +44,7 @@ def __init__(self):
         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: