Merge branch 'main-dev' of https://github.com/unum-cloud/uform into m…

…ain-dev
unum-cloud · Apr 25, 2024 · 1b821cc · 1b821cc
2 parents 4bef407 + 5ff26ad
commit 1b821cc
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(