Fix: Missing switch cases

unum-cloud · Apr 25, 2024 · 5ff26ad · 5ff26ad
1 parent 0af7cbf
commit 5ff26ad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/swift/Encoders.swift b/swift/Encoders.swift
@@ -32,7 +32,7 @@ public enum Embedding {
     /// - Returns: nil if the data type is unsupported.
     init?(from multiArray: MLMultiArray) {
         switch multiArray.dataType {
-        case .float64:
+        case .float64, .double:
             self = .f64s(
                 Array(
                     UnsafeBufferPointer(
@@ -41,7 +41,7 @@ public enum Embedding {
                     )
                 )
             )
-        case .float32:
+        case .float32, .float:
             self = .f32s(
                 Array(
                     UnsafeBufferPointer(