fix

Author: apoorvkh

src/torchrunx/launcher.py

@@ -278,7 +278,7 @@ def index(self, hostname: str, rank: int) -> Any:
 
     def rank(self, i: int) -> Any:
         """Get return value from worker by global rank."""
-        return self.by_rank()[i]
+        return self.by_ranks()[i]
 
 
 def _resolve_hostnames(hostnames: list[str] | Literal["auto", "slurm"]) -> list[str]:

tests/test_func.py

@@ -13,7 +13,7 @@ def test_launch() -> None:
         workers_per_host="slurm",
     )
 
-    result_values = result.by_rank()
+    result_values = result.by_ranks()
 
     t = True
     for i in range(len(result_values)):