From 34b4356bad189b0a9403af423b3917e0b4fd7502 Mon Sep 17 00:00:00 2001
From: Lucky Wong <lekai.huang@gmail.com>
Date: Fri, 29 Jul 2022 11:28:52 +0800
Subject: [PATCH] correction for get rank id. (#507)

* Fix no attribute 'data' error.

* minor fixes

* correction for get rank id.
---
 icefall/dist.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/icefall/dist.py b/icefall/dist.py
index 6334f9c13..7016beafb 100644
--- a/icefall/dist.py
+++ b/icefall/dist.py
@@ -57,9 +57,9 @@ def get_rank():
     if "RANK" in os.environ:
         return int(os.environ["RANK"])
     elif dist.is_available() and dist.is_initialized():
-        return dist.rank()
+        return dist.get_rank()
     else:
-        return 1
+        return 0
 
 
 def get_local_rank():