Skip to content

Commit a402faf

Browse files
committed
use len on tuple to check if empty
1 parent e8963dc commit a402faf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/accelerate/utils/dataclasses.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2999,7 +2999,7 @@ def active_mesh_dims(self):
29992999

30003000
def build_device_mesh(self, device_type: str):
30013001
mesh = self.get_mesh()
3002-
if not mesh:
3002+
if len(mesh) == 0:
30033003
return
30043004
mesh_dim_names, mesh_shape = mesh
30053005
device_mesh = torch.distributed.init_device_mesh(
@@ -3028,7 +3028,7 @@ def get_mesh(self) -> tuple[tuple[int, ...], tuple[str, ...]]:
30283028
mesh_dims.items(),
30293029
key=lambda x: (mesh_order.index(x[0])),
30303030
)
3031-
return zip(*sorted_items)
3031+
return tuple(zip(*sorted_items))
30323032

30333033
def __post_init__(self):
30343034
# Basic size validation

0 commit comments

Comments
 (0)