From f621a89ccc33513792fb741c281c8a4d68eb6247 Mon Sep 17 00:00:00 2001
From: Pedro Cuenca <pedro@huggingface.co>
Date: Wed, 14 Dec 2022 19:43:41 +0100
Subject: [PATCH] Accept suggestion from code review

Co-authored-by: Alejandro Isaza <167236+alejandro-isaza@users.noreply.github.com>
---
 swift/StableDiffusion/pipeline/Scheduler.swift | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/swift/StableDiffusion/pipeline/Scheduler.swift b/swift/StableDiffusion/pipeline/Scheduler.swift
index 6658b85..99a08e2 100644
--- a/swift/StableDiffusion/pipeline/Scheduler.swift
+++ b/swift/StableDiffusion/pipeline/Scheduler.swift
@@ -55,7 +55,7 @@ public extension Scheduler {
     /// - Returns: sum_i weights[i]*values[i]
     func weightedSum(_ weights: [Double], _ values: [MLShapedArray<Float32>]) -> MLShapedArray<Float32> {
         assert(weights.count > 1 && values.count == weights.count)
-        assert(values.allSatisfy({$0.scalarCount == values.first!.scalarCount}))
+        assert(values.allSatisfy({ $0.scalarCount == values.first!.scalarCount }))
         var w = Float(weights.first!)
         var scalars = values.first!.scalars.map({ $0 * w })
         for next in 1 ..< values.count {