Update schema.rs

synnada-ai · Nov 28, 2024 · 2a0138b · 2a0138b
1 parent 4df9695
commit 2a0138b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/arrow-schema/src/schema.rs b/arrow-schema/src/schema.rs
@@ -461,6 +461,17 @@ impl Schema {
     pub fn remove(&mut self, index: usize) -> FieldRef {
         self.fields.remove(index)
     }
+
+    /// THIS METHOD IS ARAS ONLY
+    ///
+    /// Converts the fields of the batch to nullable.
+    pub fn make_fields_nullable(&mut self) {
+        let fields = self
+            .fields
+            .iter()
+            .map(|field| field.as_ref().clone().with_nullable(true));
+        self.fields = Fields::from_iter(fields);
+    }
 }
 
 impl fmt::Display for Schema {