diff --git a/Source/MLX/MLXArray+Indexing.swift b/Source/MLX/MLXArray+Indexing.swift index fe098a38..a56f3b29 100644 --- a/Source/MLX/MLXArray+Indexing.swift +++ b/Source/MLX/MLXArray+Indexing.swift @@ -853,7 +853,7 @@ func updateSlice( var strides = [Int32](repeating: 1, count: ndim) // If it's just a simple slice, just do a slice update and return - if operations.count == 1, case let .slice(slice) = operations[0] { + if operations.count == 1, case .slice(let slice) = operations[0] { let size = src.dim(0).int32 starts[0] = slice.start(size) ends[0] = slice.end(size) diff --git a/Source/MLXNN/Module.swift b/Source/MLXNN/Module.swift index 54ade86d..2f7012a7 100644 --- a/Source/MLXNN/Module.swift +++ b/Source/MLXNN/Module.swift @@ -1582,8 +1582,9 @@ extension UpdateError: LocalizedError { "Unable to collect modules from container: \(path.joined(separator: ".")) in \(modules.joined(separator: "."))" case .mismatchedContainers(let base, let key): return "Mismatched containers: \(base) \(key)" - case let .mismatchedSize( - path, modules, expectedShape: expectedShape, actualShape: actualShape): + case .mismatchedSize( + let + path, let modules, let expectedShape, let actualShape): return "Mismatched parameter \(path.joined(separator: ".")) in \(modules.joined(separator: ".")) shape. Actual \(actualShape), expected \(expectedShape)" case .keyNotFound(let path, let modules): diff --git a/Tests/MLXTests/ModuleTests.swift b/Tests/MLXTests/ModuleTests.swift index 6d3c4962..0059223e 100644 --- a/Tests/MLXTests/ModuleTests.swift +++ b/Tests/MLXTests/ModuleTests.swift @@ -555,7 +555,7 @@ class ModuleTests: XCTestCase { verify: .all) ) { error in guard let error = error as? UpdateError, - case let .keyNotFound(path, modules) = error + case .keyNotFound(let path, let modules) = error else { XCTFail("Expected to fail with UpdateError.keyNotFound, but got: \(error)") return @@ -586,8 +586,10 @@ class ModuleTests: XCTestCase { verify: .all) ) { error in guard let error = error as? UpdateError, - case let .mismatchedSize( - path, modules, expectedShape: expectedShape, actualShape: actualShape) = + case .mismatchedSize( + let + path, let modules, expectedShape: let expectedShape, + actualShape: let actualShape) = error else { XCTFail("Expected to fail with UpdateError.mismatchedSize, but got: \(error)")