add more supertypes

pola-rs · Dec 23, 2021 · 3c85adf · 3c85adf
1 parent fc1c295
commit 3c85adf
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/polars/polars-core/src/utils/mod.rs b/polars/polars-core/src/utils/mod.rs
@@ -541,6 +541,10 @@ fn _get_supertype(l: &DataType, r: &DataType) -> Option<DataType> {
         (UInt32, Float32) => Some(Float32),
         (UInt32, Float64) => Some(Float64),
         (UInt32, Boolean) => Some(UInt32),
+        #[cfg(feature = "dtype-date")]
+        (UInt32, Date) => Some(Int64),
+        #[cfg(feature = "dtype-datetime")]
+        (UInt32, Datetime) => Some(Int64),
 
         (UInt64, UInt8) => Some(UInt64),
         (UInt64, UInt16) => Some(UInt64),
@@ -612,6 +616,10 @@ fn _get_supertype(l: &DataType, r: &DataType) -> Option<DataType> {
         (Float64, Time) => Some(Float64),
         (Float64, Boolean) => Some(Float64),
 
+        #[cfg(feature = "dtype-datetime")]
+        (Date, UInt32) => Some(Int64),
+        #[cfg(feature = "dtype-datetime")]
+        (Date, UInt64) => Some(Int64),
         #[cfg(feature = "dtype-datetime")]
         (Date, Int32) => Some(Int32),
         #[cfg(feature = "dtype-datetime")]
@@ -623,6 +631,10 @@ fn _get_supertype(l: &DataType, r: &DataType) -> Option<DataType> {
         #[cfg(feature = "dtype-datetime")]
         (Date, Datetime) => Some(Datetime),
 
+        #[cfg(feature = "dtype-date")]
+        (Datetime, UInt32) => Some(Int64),
+        #[cfg(feature = "dtype-date")]
+        (Datetime, UInt64) => Some(Int64),
         #[cfg(feature = "dtype-date")]
         (Datetime, Int32) => Some(Int64),
         #[cfg(feature = "dtype-date")]