fix

Signed-off-by: Joe Isaacs <joe.isaacs@live.co.uk>

Author: joseph-isaacs

vortex-array/src/arrow/record_batch.rs

@@ -29,7 +29,7 @@ impl TryFrom<&ArrayRef> for RecordBatch {
         };
 
         vortex_ensure!(
-            matches!(struct_array.validity()?, Validity::AllValid),
+            struct_array.validity()?.no_nulls(),
             "RecordBatch can only be constructed from StructArray with no nulls"
         );
 

vortex-array/src/validity.rs

@@ -112,6 +112,13 @@ impl Validity {
         }
     }
 
+    /// Returns `true` if this validity guarantees no null values, i.e. it is either
+    /// [`Validity::NonNullable`] or [`Validity::AllValid`].
+    #[inline]
+    pub fn no_nulls(&self) -> bool {
+        matches!(self, Self::NonNullable | Self::AllValid)
+    }
+
     /// The union nullability and validity.
     #[inline]
     pub fn union_nullability(self, nullability: Nullability) -> Self {