address comment

Author: klion26

parquet-variant-compute/src/variant_array.rs

@@ -866,51 +866,42 @@ fn typed_value_to_variant(typed_value: &ArrayRef, index: usize) -> Variant<'_, '
         DataType::Float64 => {
             primitive_conversion_single_value!(Float64Type, typed_value, index)
         }
-        DataType::Timestamp(timeunit, tz) => {
-            match (timeunit, tz) {
-                (TimeUnit::Microsecond, Some(_)) => {
-                    generic_conversion_single_value!(
-                        TimestampMicrosecondType,
-                        as_primitive,
-                        |v| DateTime::from_timestamp_micros(v).unwrap(),
-                        typed_value,
-                        index
-                    )
-                }
-                (TimeUnit::Microsecond, None) => {
-                    generic_conversion_single_value!(
-                        TimestampMicrosecondType,
-                        as_primitive,
-                        |v| DateTime::from_timestamp_micros(v).unwrap().naive_utc(),
-                        typed_value,
-                        index
-                    )
-                }
-                (TimeUnit::Nanosecond, Some(_)) => {
-                    generic_conversion_single_value!(
-                        TimestampNanosecondType,
-                        as_primitive,
-                        DateTime::from_timestamp_nanos,
-                        typed_value,
-                        index
-                    )
-                }
-                (TimeUnit::Nanosecond, None) => {
-                    generic_conversion_single_value!(
-                        TimestampNanosecondType,
-                        as_primitive,
-                        |v| DateTime::from_timestamp_nanos(v).naive_utc(),
-                        typed_value,
-                        index
-                    )
-                }
-                // Variant timestamp only support time unit with microsecond or nanosecond precision
-                _ => panic!(
-                    "Variant only support timestamp with microsecond or nanosecond precision"
-                ),
-            }
+        DataType::Timestamp(TimeUnit::Microsecond, Some(_)) => {
+            generic_conversion_single_value!(
+                TimestampMicrosecondType,
+                as_primitive,
+                |v| DateTime::from_timestamp_micros(v).unwrap(),
+                typed_value,
+                index
+            )
+        }
+        DataType::Timestamp(TimeUnit::Microsecond, None) => {
+            generic_conversion_single_value!(
+                TimestampMicrosecondType,
+                as_primitive,
+                |v| DateTime::from_timestamp_micros(v).unwrap().naive_utc(),
+                typed_value,
+                index
+            )
+        }
+        DataType::Timestamp(TimeUnit::Nanosecond, Some(_)) => {
+            generic_conversion_single_value!(
+                TimestampNanosecondType,
+                as_primitive,
+                DateTime::from_timestamp_nanos,
+                typed_value,
+                index
+            )
+        }
+        DataType::Timestamp(TimeUnit::Nanosecond, None) => {
+            generic_conversion_single_value!(
+                TimestampNanosecondType,
+                as_primitive,
+                |v| DateTime::from_timestamp_nanos(v).naive_utc(),
+                typed_value,
+                index
+            )
         }
-
         // todo other types here (note this is very similar to cast_to_variant.rs)
         // so it would be great to figure out how to share this code
         _ => {