Fix nested guard for Series::try_from arrow2 arrays

Eventual-Inc · Aug 19, 2023 · 7e2ffb0 · 7e2ffb0
1 parent 7517df1
commit 7e2ffb0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/daft-core/src/series/from.rs b/src/daft-core/src/series/from.rs
@@ -24,7 +24,11 @@ impl TryFrom<(&str, Box<dyn arrow2::array::Array>)> for Series {
         // Corner-case nested logical types that have not yet been migrated to new Array formats
         // to hold only casted physical arrow arrays.
         let physical_type = dtype.to_physical();
-        if physical_type != dtype {
+        if (matches!(dtype, DataType::List(..))
+            || matches!(dtype, DataType::Struct(..))
+            || dtype.is_extension())
+            && physical_type != dtype
+        {
             let arrow_physical_type = physical_type.to_arrow()?;
             let casted_array = arrow2::compute::cast::cast(
                 array.as_ref(),