fix: enable arrow conversion from Int96 (#1653)

DrakeLin · nicklan · web-flow · commit f218eef7dafc · 2026-01-22T10:42:29.000-08:00
Co-authored-by: Nick Lanham &lt;nicklan@users.noreply.github.com&gt;
diff --git a/kernel/src/engine/arrow_conversion.rs b/kernel/src/engine/arrow_conversion.rs
@@ -246,6 +246,12 @@ impl TryFromArrow<&ArrowDataType> for DataType {
             {
                 Ok(DataType::TIMESTAMP)
             }
+            ArrowDataType::Timestamp(TimeUnit::Nanosecond, None) => Ok(DataType::TIMESTAMP_NTZ),
+            ArrowDataType::Timestamp(TimeUnit::Nanosecond, Some(tz))
+                if tz.eq_ignore_ascii_case("utc") =>
+            {
+                Ok(DataType::TIMESTAMP)
+            }
             ArrowDataType::Struct(fields) => DataType::try_struct_type_from_results(
                 fields.iter().map(|field| field.as_ref().try_into_kernel()),
             )

Original file line number	Diff line number	Diff line change
`@@ -246,6 +246,12 @@ impl TryFromArrow<&ArrowDataType> for DataType {`
`246`	`246`	`{`
`247`	`247`	`Ok(DataType::TIMESTAMP)`
`248`	`248`	`}`
	`249`	`+ ArrowDataType::Timestamp(TimeUnit::Nanosecond, None) => Ok(DataType::TIMESTAMP_NTZ),`
	`250`	`+ ArrowDataType::Timestamp(TimeUnit::Nanosecond, Some(tz))`
	`251`	`+ if tz.eq_ignore_ascii_case("utc") =>`
	`252`	`+ {`
	`253`	`+ Ok(DataType::TIMESTAMP)`
	`254`	`+ }`
`249`	`255`	`ArrowDataType::Struct(fields) => DataType::try_struct_type_from_results(`
`250`	`256`	`fields.iter().map(\|field\| field.as_ref().try_into_kernel()),`
`251`	`257`	`)`