Skip to content

Commit 9bbae71

Browse files
authored
Fixed failing mypy on python 3.13 (#3464)
1 parent 1c20a76 commit 9bbae71

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ignite/distributed/comp_models/native.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def _compute_node_and_local_ranks(rank: int, hostnames: List[Tuple[str, ...]]) -
178178
c: Counter = Counter(hostnames)
179179
sizes = torch.tensor([0] + list(c.values()))
180180
cumsum_sizes = torch.cumsum(sizes, dim=0)
181-
node_rank = (rank // cumsum_sizes[1:]).clamp(0, 1).sum().item()
181+
node_rank = cast(int, (rank // cumsum_sizes[1:]).clamp(0, 1).sum().item())
182182
local_rank = rank - cumsum_sizes[node_rank].item()
183183
return int(local_rank), node_rank
184184

0 commit comments

Comments
 (0)