RecordBatch::try_new

Author: nevi-me

src/dataframe.rs

@@ -135,7 +135,8 @@ impl DataFrame {
 
         arrays.into_iter().for_each(|array| {
             dbg!(array.len());
-            batches.push(RecordBatch::new(self.schema.clone(), array));
+            // the unwrap is infallible as we're passing data that's already been verified
+            batches.push(RecordBatch::try_new(self.schema.clone(), array).unwrap());
         });
 
         batches

src/io/feather.rs

@@ -386,7 +386,7 @@ impl<R: Read + Seek> FeatherReader<R> {
             };
         }
 
-        Ok(RecordBatch::new(Arc::new(Schema::new(fields)), arrays))
+        Ok(RecordBatch::try_new(Arc::new(Schema::new(fields)), arrays)?)
     }
 }