Handle pushing None in Dense unions

DataEngineeringLabs · May 8, 2023 · c35d769 · c35d769
1 parent bef07b9
commit c35d769
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/arrow2_convert_derive/src/derive_enum.rs b/arrow2_convert_derive/src/derive_enum.rs
@@ -119,7 +119,6 @@ pub fn expand_serialize(input: DeriveEnum) -> TokenStream {
         ..
     } = (&input).into();
 
-    let first_variant = &variant_names[0];
     let is_dense = input.is_dense;
 
     let mutable_array_name = &input.common.mutable_array_name();
@@ -227,6 +226,7 @@ pub fn expand_serialize(input: DeriveEnum) -> TokenStream {
         let first_name = &variant_names[0];
         quote! {
             self.types.push(0);
+            self.offsets.push((self.#first_name.len()) as i32);
             <#first_array_type as MutableArray>::push_null(&mut self.#first_name);
         }
     } else {