Skip to content

Commit c6efbfd

Browse files
committed
Swift lint
1 parent 8e253e9 commit c6efbfd

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

Source/Examples/DistributedWorker.swift

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -920,7 +920,8 @@ struct DistributedWorker {
920920
"weight_f32": MLXArray(
921921
rank == 0 ? [2.0, 4.0] as [Float] : [4.0, 8.0] as [Float]),
922922
"weight_f16": MLXArray(
923-
rank == 0 ? [10.0, 20.0] as [Float] : [30.0, 40.0] as [Float]).asType(.float16),
923+
rank == 0 ? [10.0, 20.0] as [Float] : [30.0, 40.0] as [Float]
924+
).asType(.float16),
924925
]
925926
let mixedGrads = ModuleParameters.unflattened(mixedFlat)
926927
let mixedResult = averageGradients(gradients: mixedGrads, group: group)

0 commit comments

Comments
 (0)