Impl serieslike for Decimal128Array, Int128Array; add Decimal128 to L…

…ogical match macro
Eventual-Inc · Jul 10, 2023 · 409d2c2 · 409d2c2
1 parent 1c34316
commit 409d2c2
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/src/daft-core/src/datatypes/matching.rs b/src/daft-core/src/datatypes/matching.rs
@@ -230,6 +230,7 @@ macro_rules! with_match_daft_logical_types {(
     use $crate::datatypes::*;
 
     match $key_type {
+        Decimal128(..) => __with_ty__! { Decimal128Type },
         Date => __with_ty__! { DateType },
         Duration(..) => __with_ty__! { DurationType },
         Timestamp(..) => __with_ty__! { TimestampType },

diff --git a/src/daft-core/src/series/array_impl/data_array.rs b/src/daft-core/src/series/array_impl/data_array.rs
@@ -14,8 +14,8 @@ use crate::series::Field;
 use crate::{
     datatypes::{
         BinaryArray, BooleanArray, ExtensionArray, FixedSizeListArray, Float32Array, Float64Array,
-        Int16Array, Int32Array, Int64Array, Int8Array, ListArray, NullArray, StructArray,
-        UInt16Array, UInt32Array, UInt64Array, UInt8Array, Utf8Array,
+        Int128Array, Int16Array, Int32Array, Int64Array, Int8Array, ListArray, NullArray,
+        StructArray, UInt16Array, UInt32Array, UInt64Array, UInt8Array, Utf8Array,
     },
     series::series_like::SeriesLike,
     with_match_integer_daft_types,
@@ -327,6 +327,7 @@ impl_series_like_for_data_array!(Int8Array);
 impl_series_like_for_data_array!(Int16Array);
 impl_series_like_for_data_array!(Int32Array);
 impl_series_like_for_data_array!(Int64Array);
+impl_series_like_for_data_array!(Int128Array);
 impl_series_like_for_data_array!(UInt8Array);
 impl_series_like_for_data_array!(UInt16Array);
 impl_series_like_for_data_array!(UInt32Array);

diff --git a/src/daft-core/src/series/array_impl/logical_array.rs b/src/daft-core/src/series/array_impl/logical_array.rs
@@ -1,5 +1,6 @@
 use crate::datatypes::logical::{
-    DateArray, DurationArray, EmbeddingArray, FixedShapeImageArray, ImageArray, TimestampArray,
+    DateArray, Decimal128Array, DurationArray, EmbeddingArray, FixedShapeImageArray, ImageArray,
+    TimestampArray,
 };
 use crate::datatypes::BooleanArray;
 
@@ -197,6 +198,7 @@ macro_rules! impl_series_like_for_logical_array {
     };
 }
 
+impl_series_like_for_logical_array!(Decimal128Array);
 impl_series_like_for_logical_array!(DateArray);
 impl_series_like_for_logical_array!(DurationArray);
 impl_series_like_for_logical_array!(EmbeddingArray);