[BE] follow autoformating and linter

XilunWu · XilunWu · commit 0ec95a6c4c09 · 2025-04-23T23:15:54.000-07:00
ghstack-source-id: f757813 Pull Request resolved: pytorch/pytorch#151507
diff --git a/torch/distributed/tensor/_ops/_view_ops.py b/torch/distributed/tensor/_ops/_view_ops.py
@@ -560,9 +560,11 @@ def get_in_dim_to_shard(cmd: DimSpec) -> Optional[InputDim]:
             shard_dim_map[in_dim.input_dim] = dim
 
     input_tgt_placements = [
-        Replicate()
-        if isinstance(p, Shard) and not shardable_dims[p.dim][mesh_dim]
-        else p
+        (
+            Replicate()
+            if isinstance(p, Shard) and not shardable_dims[p.dim][mesh_dim]
+            else p
+        )
         for mesh_dim, p in enumerate(input_src_placements)
     ]
     output_placements = [
