fix lint check

pytorch · Jun 7, 2023 · 0afebac · 0afebac
1 parent ef61de3
commit 0afebac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/distributed/_tensor/random.py b/torch/distributed/_tensor/random.py
@@ -7,7 +7,7 @@
 import torch.distributed as dist
 
 from torch import Tensor
-from torch.distributed._tensor.device_mesh import DeviceMesh, _get_device_handle
+from torch.distributed._tensor.device_mesh import _get_device_handle, DeviceMesh
 from torch.distributed._tensor.placement_types import DTensorSpec, Shard